diff --git a/.github/workflows/test_package_build.yml b/.github/workflows/test_package_build.yml
index 22b3e5d..6eb09b5 100644
--- a/.github/workflows/test_package_build.yml
+++ b/.github/workflows/test_package_build.yml
@@ -78,7 +78,11 @@ jobs:
       - name: Install test extras
         run: pip install package_name[test]
       - name: Run tests
-        run: pytest --doctest-modules -v --pyargs track_linearization
+        run: pytest --doctest-modules -v --pyargs package_name
+      - name: Test notebook
+        shell: bash -l {0}
+        run: |
+          jupyter nbconvert --to notebook --ExecutePreprocessor.kernel_name=python3 --execute notebooks/test_linearization.ipynb
   publish:
     runs-on: ubuntu-latest
     needs: [test-package]