diff --git a/.github/workflows/testing-and-coverage.yml b/.github/workflows/testing-and-coverage.yml
index 0a761f4..d2bcff2 100644
--- a/.github/workflows/testing-and-coverage.yml
+++ b/.github/workflows/testing-and-coverage.yml
@@ -25,6 +25,6 @@ jobs:
         python-version: ${{ matrix.python-version }}
     - name: Run unit tests with pytest
       run: |
-        python -m pytest tests --cov=autora --cov-report=xml
+        python -m pytest tests/* --cov=autora --cov-report=xml
     - name: Upload coverage report to codecov
       uses: codecov/codecov-action@v3
diff --git a/pyproject.toml b/pyproject.toml
index 0f0f11d..e6db6a3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -43,6 +43,7 @@ dev = [
     "ipython", # Also used in building notebooks into Sphinx
     "matplotlib", # Used in sample notebook intro_notebook.ipynb
     "numpy", # Used in sample notebook intro_notebook.ipynb
+    "ipykernel",
 ]
 train = [
     "mlflow",