diff --git a/.github/workflows/publish-to-docker.yml b/.github/workflows/publish-to-docker.yml
index 81ba0ca9d..ae1385d55 100644
--- a/.github/workflows/publish-to-docker.yml
+++ b/.github/workflows/publish-to-docker.yml
@@ -44,7 +44,7 @@ jobs:
             # Function to check if version exists in a repository
             check_version() {
                 local repo=$1
-                local status_code=$(curl -s -o /dev/null -w "%{http_code}" "https://$repo.org/pypi/llama-stack/${{ inputs.version }}")
+                local status_code=$(curl -s -o /dev/null -w "%{http_code}" "https://$repo.org/pypi/llama-stack/${{ steps.version.outputs.version }}")
                 return $([ "$status_code" -eq 200 ])
             }