diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
index ad1937724..b0854d40b 100644
--- a/.buildkite/pipeline.yml
+++ b/.buildkite/pipeline.yml
@@ -34,7 +34,7 @@ steps:
     agents:
       queue: mithril-h100-pool
     commands:
-      - pytest -s -v tests/single_stage/test_diffusion_model.py
+      - pytest -s -v tests/single_stage/test_diffusion_model.py # update this to your actual test file
     plugins:
       - kubernetes:
           podSpec: