diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index f7a1270..5d78b4c 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -30,7 +30,7 @@ jobs:
           version: "latest"
           use-cache: true
       - name: Run Release target
-        run: earthly --ci --output +release
+        run: earthly --ci --output +release --VERSION=${{ github.ref }}
       - name: Release
         uses: softprops/action-gh-release@v2
         if: startsWith(github.ref, 'refs/tags/')
diff --git a/Earthfile b/Earthfile
index 941e08e..3d88c45 100644
--- a/Earthfile
+++ b/Earthfile
@@ -76,12 +76,14 @@ package-tar:
 
 install-script:
     FROM +ubuntu
-    
-    ARG VERSION=$(head -n 1 PE_VERSION)
+
+    ARG VERSION
+    ARG PE_VERSION=$(head -n 1 PE_VERSION)
     ARG IMAGE_REPO=${SPECTRO_PUB_REPO}/edge
-    ARG AGENT_URL_PREFIX=https://github.com/spectrocloud/agent-mode/releases/latest/download
+    # https://github.com/spectrocloud/agent-mode/releases/download/v4.5.0-rc2/palette-agent-linux-amd64
+    ARG AGENT_URL_PREFIX=https://github.com/spectrocloud/agent-mode/releases/download/${VERSION}
     
-    ENV PE_VERSION=${VERSION}
+    ENV PE_VERSION=${PE_VERSION}
     ENV IMAGE_REPO=${IMAGE_REPO}
     ENV AGENT_URL_PREFIX=${AGENT_URL_PREFIX}