diff --git a/.github/workflows/stateful_dataloader_ci.yml b/.github/workflows/stateful_dataloader_ci.yml
index ca8dac59c..a62919527 100644
--- a/.github/workflows/stateful_dataloader_ci.yml
+++ b/.github/workflows/stateful_dataloader_ci.yml
@@ -83,4 +83,4 @@ jobs:
           cd ..
       - name: Run StatefulDataLoader tests with pytest
         if: ${{ ! contains(github.event.pull_request.labels.*.name, 'ciflow/slow') }}
-        run: pytest --durations=0 --no-header -v test/stateful_dataloader/test_dataloader.py
+        run: pytest --durations=0 --no-header -v test/stateful_dataloader/test_state_dict.py