diff --git a/.github/workflows/nix_build.yaml b/.github/workflows/nix_build.yaml
index 33c8ff143..5a10bd39c 100644
--- a/.github/workflows/nix_build.yaml
+++ b/.github/workflows/nix_build.yaml
@@ -47,7 +47,7 @@ jobs:
         if [ "${{ github.event_name }}" = "pull_request" ]; then
           export TAG=nix-sha-${{ env.GITHUB_SHA_SHORT }}
         else
-          export TAG=nix-{{ version }}
+          export TAG=nix-${{ github.ref_name }}
         fi
         export IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:$TAG
         nix-shell -p skopeo --command "skopeo --insecure-policy copy docker-archive:$(readlink -f ./result) docker://$IMAGE --dest-compress-format zstd"