diff --git a/.github/workflows/docker-unified.yml b/.github/workflows/docker-unified.yml
index 6d0e4213bfae8..54160b71d7864 100644
--- a/.github/workflows/docker-unified.yml
+++ b/.github/workflows/docker-unified.yml
@@ -434,6 +434,7 @@ jobs:
         elasticsearch_setup_build,
         mae_consumer_build,
         mce_consumer_build,
+        datahub_upgrade_build,
       ]
     steps:
       - name: Check out the repo
@@ -487,6 +488,11 @@ jobs:
         if: ${{ needs.setup.outputs.publish != 'true' }}
         with:
           image: ${{ env.DATAHUB_MAE_CONSUMER_IMAGE }}:${{ needs.setup.outputs.unique_tag }}
+      - name: Download upgrade image
+        uses: ishworkh/docker-image-artifact-download@v1
+        if: ${{ needs.setup.outputs.publish != 'true' }}
+        with:
+          image: ${{ env.DATAHUB_UPGRADE_IMAGE }}:${{ needs.setup.outputs.unique_tag }}
       - name: Calculate pip install plan
         id: "packages_checksum"
         run: |