diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 44938aea..d65037ab 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -195,7 +195,7 @@ jobs:
           # export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
           export DOCKER_DEVICES=/dev/kfd,/dev/dri
           # export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}"
-          # export HF_TOKEN=${{ secrets.HF_TOKEN }}
+          export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
           # python -c 'import docker; client = docker.from_env(); client.containers.run("registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-318bbc7-rocm", devices=["/dev/kfd", "/dev/dri"], env={"MODEL_ID": "gpt2"}, detach=True)'
           which pytest
diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py
index 319ba8d3..4633f33f 100644
--- a/integration-tests/conftest.py
+++ b/integration-tests/conftest.py
@@ -533,7 +533,7 @@ def launcher(event_loop):
             DOCKER_IMAGE,
             command=args,
             # name=container_name,
-            # environment=env,
+            environment=env,
             # auto_remove=False,
             # detach=True,
             # device_requests=device_requests,