diff --git a/Dockerfile b/Dockerfile
index 9c0eb20..8eecc42 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,8 +8,9 @@ ADD package.json /crc-bpmn/
 
 COPY . /crc-bpmn/
 
+ARG build_config=staging
 RUN npm install && \
-    npm run build:${APP_ENVIRONMENT}
+    npm run build:$build_config
 
 ### STAGE 2: Run ###
 FROM nginx:alpine
diff --git a/deploy.sh b/deploy.sh
index 3c741d1..012097e 100755
--- a/deploy.sh
+++ b/deploy.sh
@@ -7,7 +7,7 @@ export PATH=$PATH:$HOME/.local/bin;
 # Build and push Docker image to Docker Hub
 echo "$DOCKER_TOKEN" | docker login -u "$DOCKER_USERNAME" --password-stdin;
 docker build \
-  --build-arg APP_ENVIRONMENT="$APP_ENVIRONMENT" \
+  --build-arg build_config="$DEPLOY_BUILD_CONFIG" \
   --no-cache -t sartography/cr-connect-bpmn:latest . \
   || exit 1;
 docker push sartography/cr-connect-bpmn:latest || exit 1;