diff --git a/packages/cp-gql-mock-server/Dockerfile b/packages/cp-gql-mock-server/Dockerfile
index c3ac7dc7..a4404fee 100644
--- a/packages/cp-gql-mock-server/Dockerfile
+++ b/packages/cp-gql-mock-server/Dockerfile
@@ -8,5 +8,7 @@ RUN npm install lerna@^2.0.0-rc.5 \
 WORKDIR /home/node/app/packages/cp-gql-mock-server
 COPY packages/cp-gql-mock-server/etc/containerpilot.json /etc/
 
+ARG CIRCLE_BRANCH
+
 EXPOSE 3000
 CMD ["/bin/containerpilot", "yarn", "run", "start"]
diff --git a/packages/ui-toolkit/Dockerfile b/packages/ui-toolkit/Dockerfile
index e1496fd3..ea066feb 100644
--- a/packages/ui-toolkit/Dockerfile
+++ b/packages/ui-toolkit/Dockerfile
@@ -8,5 +8,7 @@ RUN npm install lerna@^2.0.0-rc.5 \
 WORKDIR /home/node/app/packages/ui-toolkit
 COPY packages/ui-toolkit/etc/containerpilot.json /etc/
 
+ARG CIRCLE_BRANCH
+
 EXPOSE 6060
 CMD ["/bin/containerpilot", "yarn", "run", "styleguide"]