diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 6f344f77..9b9da1d7 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -48,7 +48,7 @@ jobs:
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Test with pytest and generate coverage
       run: |
-        pytest --doctest-modules pytorch_widedeep/models --cov=pytorch_widedeep/models --cov-report=xml --cov-report=term --disable-pytest-warnings tests/test_model_functioning
+        pytest --doctest-modules pytorch_widedeep --cov=pytorch_widedeep --cov-report=xml --cov-report=term --disable-pytest-warnings tests
     - name: Upload coverage to Codecov
       uses: codecov/codecov-action@v4
       with: