diff --git a/.github/workflows/publish-test-images.yaml b/.github/workflows/publish-test-images.yaml
index 72fb10bfd337a40e5c7ecfeb7dde8aba14b21a1e..b378f70289c107fb328691488e30dfcfc0c95702 100644
--- a/.github/workflows/publish-test-images.yaml
+++ b/.github/workflows/publish-test-images.yaml
@@ -40,8 +40,8 @@ jobs:
         working-directory: tests/docker
         run: |
           docker-compose pull --ignore-pull-failures pytest
-          TAG=${{ steps.extracter.outputs.version }}-${{ steps.extracter.outputs.sha_short }} docker-compose build pytest
-          TAG=latest docker-compose build pytest
+          IMAGE_TAG=${{ steps.extracter.outputs.version }}-${{ steps.extracter.outputs.sha_short }} docker-compose build pytest
+          IMAGE_TAG=latest docker-compose build pytest
       - name: Docker Push
         if: success() && github.event_name == 'push' && github.repository == 'milvus-io/milvus'
         continue-on-error: true
@@ -50,6 +50,6 @@ jobs:
         run: |
           docker login -u ${{ secrets.DOCKERHUB_USER }} \
                        -p ${{ secrets.DOCKERHUB_TOKEN }}
-          TAG=${{ steps.extracter.outputs.version }}-${{ steps.extracter.outputs.sha_short }} docker-compose push pytest
-          TAG=latest docker-compose push pytest
+          IMAGE_TAG=${{ steps.extracter.outputs.version }}-${{ steps.extracter.outputs.sha_short }} docker-compose push pytest
+          IMAGE_TAG=latest docker-compose push pytest
           echo "Push pytest image Succeeded"