diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 6135cca4..ced2eb27 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -22,7 +22,7 @@ jobs:
           pip install -e .
     - name: Generate coverage report
       run: |
-        pytest test/unit/test_tutorials/test_rec_system.py --cov=./ --cov-report=xml
+        pytest test/unit/with_pytorch/test_agent.py --cov=./ --cov-report=xml
     - name: Upload coverage to Codecov
       if: success() || failure()
       uses: codecov/codecov-action@v4