diff --git a/buildkite/test-template-ci.j2 b/buildkite/test-template-ci.j2
index 62044d93..9f8dbef9 100644
--- a/buildkite/test-template-ci.j2
+++ b/buildkite/test-template-ci.j2
@@ -63,6 +63,9 @@ plugins:
         - VLLM_USAGE_SOURCE=ci-test
         - HF_HOME={{ hf_home_fsx }}
         - HF_TOKEN
+        {% if step.torch_nightly == "1" %}
+        - TORCH_NIGHTLY=1
+        {% endif %}
         {% if branch == "main" %}
         - BUILDKITE_ANALYTICS_TOKEN
         {% endif %}
@@ -315,7 +318,7 @@ steps:
       {% else %}
         {% set ns.blocked = 0 %}
       {% endif %}
-      
+
       {% if ns.blocked == 1 or (step.optional and nightly != "1") %}
       - block: "Run Torch Nightly {{ step.label }}"
         depends_on: image-build-torch-nightly