diff --git a/docker/datahub-upgrade/datahub-upgrade.sh b/docker/datahub-upgrade/datahub-upgrade.sh
index 698b5c2537a94..8cbe8f8fd27c3 100755
--- a/docker/datahub-upgrade/datahub-upgrade.sh
+++ b/docker/datahub-upgrade/datahub-upgrade.sh
@@ -1,4 +1,4 @@
 #!/bin/bash
 
 DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
-cd $DIR && docker pull acryldata/datahub-upgrade:latest && docker run --env-file ./env/docker.env --network="datahub_network" acryldata/datahub-upgrade:latest "$@"
\ No newline at end of file
+cd $DIR && docker pull acryldata/datahub-upgrade:head && docker run --env-file ./env/docker.env --network="datahub_network" acryldata/datahub-upgrade:latest "$@"
\ No newline at end of file
diff --git a/docker/datahub-upgrade/env/docker.env b/docker/datahub-upgrade/env/docker.env
index 7f9915e7e12f0..462a6d851ad63 100644
--- a/docker/datahub-upgrade/env/docker.env
+++ b/docker/datahub-upgrade/env/docker.env
@@ -21,9 +21,6 @@ DATAHUB_GMS_PORT=8080
 DATAHUB_MAE_CONSUMER_HOST=datahub-mae-consumer
 DATAHUB_MAE_CONSUMER_PORT=9091
 
-DATAHUB_MAE_CONSUMER_PORT=9091
-DATAHUB_MAE_CONSUMER_HOST=datahub-mae-consumer
-
 # Uncomment and set these to support SSL connection to Elasticsearch
 # ELASTICSEARCH_USE_SSL=
 # ELASTICSEARCH_SSL_PROTOCOL=
diff --git a/docs/advanced/no-code-upgrade.md b/docs/advanced/no-code-upgrade.md
index e7ff4c4ec5395..c3d11ea2236e1 100644
--- a/docs/advanced/no-code-upgrade.md
+++ b/docs/advanced/no-code-upgrade.md
@@ -42,7 +42,7 @@ It is important that the following containers are pulled and deployed simultaneo
 From the `docker` directory:
 
 ```aidl
-docker-compose down && docker-compose pull && docker-compose -p datahub up --force-recreate
+docker-compose down --remove-orphans && docker-compose pull && docker-compose -p datahub up --force-recreate
 ```
 
 #### Helm