diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index faccd16..c7ed324 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -60,6 +60,9 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
+      - name: Extract version from github ref
+        run: echo "RELEASE_VERSION=${GITHUB_REF#refs/*/}" >> $GITHUB_ENV
+
       - name: Download build artifacts
         uses: actions/download-artifact@v3
         with:
@@ -85,6 +88,8 @@ jobs:
           push: true
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
+          build-args: |
+            IMAGE_VERSION=${{ env.RELEASE_VERSION }}
 
   helm-publish:
     needs:
@@ -102,12 +107,6 @@ jobs:
           name: helm-chart
           path: helm
 
-      - name: Print file contents
-        run: | 
-          ls
-          ls helm
-          cat helm/Chart.yaml
-
       - name: Configure AWS credentials
         uses: aws-actions/configure-aws-credentials@v4
         with: