diff --git a/.github/workflows/04-code-checks.yml b/.github/workflows/04-code-checks.yml
index 96bbc5d..e01c29e 100644
--- a/.github/workflows/04-code-checks.yml
+++ b/.github/workflows/04-code-checks.yml
@@ -25,7 +25,7 @@ jobs:
 
       - name: Run pytest tests
         run: |
-          pytest.py
+          train_test.py
 
       - name: Check test results and provide merge status
         run: |