diff --git a/.github/workflows/spark-smoke-test.yml b/.github/workflows/spark-smoke-test.yml
index ac411d812deea..289b4274055c3 100644
--- a/.github/workflows/spark-smoke-test.yml
+++ b/.github/workflows/spark-smoke-test.yml
@@ -26,8 +26,33 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
+  setup:
+    runs-on: ubuntu-latest
+    outputs:
+      tag: ${{ steps.tag.outputs.tag }}
+      slim_tag: ${{ steps.tag.outputs.slim_tag }}
+      full_tag: ${{ steps.tag.outputs.full_tag }}
+      unique_tag: ${{ steps.tag.outputs.unique_tag }}
+      unique_slim_tag: ${{ steps.tag.outputs.unique_slim_tag }}
+      unique_full_tag: ${{ steps.tag.outputs.unique_full_tag }}
+      publish: ${{ steps.publish.outputs.publish }}
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v3
+      - name: Compute Tag
+        id: tag
+        run: |
+          source .github/scripts/docker_helpers.sh
+          echo "tag=$(get_tag)" >> $GITHUB_OUTPUT
+          echo "slim_tag=$(get_tag)-slim" >> $GITHUB_OUTPUT
+          echo "full_tag=$(get_tag)-full" >> $GITHUB_OUTPUT
+          echo "unique_tag=$(get_unique_tag)" >> $GITHUB_OUTPUT
+          echo "unique_slim_tag=$(get_unique_tag)-slim" >> $GITHUB_OUTPUT
+          echo "unique_full_tag=$(get_unique_tag)-full" >> $GITHUB_OUTPUT
+
   spark-smoke-test:
     runs-on: ubuntu-latest
+    needs: setup
     steps:
       - uses: actions/checkout@v3
       - name: Set up JDK 11
@@ -43,6 +68,8 @@ jobs:
       - name: Remove images
         run: docker image prune -a -f || true
       - name: Smoke test
+        env:
+          DATAHUB_VERSION: ${{ needs.setup.outputs.unique_tag }}
         run: |
           ./gradlew :metadata-integration:java:spark-lineage:integrationTest \
              -x :datahub-web-react:yarnTest \