diff --git a/.github/workflows/image-publish.yml b/.github/workflows/image-publish.yml
index c4117d4e..a4f8cd8a 100644
--- a/.github/workflows/image-publish.yml
+++ b/.github/workflows/image-publish.yml
@@ -89,7 +89,7 @@ jobs:
           cache-to: type=gha,mode=max
           build-args: |
             LATEST_RELEASE=${{ env.LATEST_RELEASE }}
-            CODEGATE_VERSION=${{ steps.version-string.outputs.tag }}
+            CODEGATE_VERSION=${{ github.ref_name }}
       - name: Capture Image Digest
         id: image-digest
         run: |