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"
diff --git a/tgi-entrypoint.sh b/tgi-entrypoint.sh
index cd551ed5e..32eccea55 100755
--- a/tgi-entrypoint.sh
+++ b/tgi-entrypoint.sh
@@ -2,5 +2,5 @@
 
 ldconfig 2>/dev/null || echo 'unable to refresh ld cache, not a big deal in most cases'
 
-source ./.venv/bin/activate
+source /usr/src/.venv/bin/activate
 exec text-generation-launcher $@