diff --git a/.azure/azure-ci-build-pipeline.yaml b/.azure/azure-ci-build-pipeline.yaml
index aea94807..7ac50ea8 100644
--- a/.azure/azure-ci-build-pipeline.yaml
+++ b/.azure/azure-ci-build-pipeline.yaml
@@ -126,3 +126,8 @@ stages:
             inputs:
               codeCoverageTool: Cobertura
               summaryFileLocation: '$(System.DefaultWorkingDirectory)/**/coverage.xml'
+          # Run the notebooks to make sure that they work
+          - script: |
+              pytest --nbmake -n=auto  src/primaite/notebooks
+              pytest --nbmake -n=auto src/primaite/simulator/_package_data
+            displayName: 'Run notebooks'
diff --git a/pyproject.toml b/pyproject.toml
index 9d53e961..badb1557 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -37,7 +37,9 @@ dependencies = [
     "typer[all]==0.9.0",
     "pydantic==2.7.0",
     "ipywidgets",
-    "deepdiff"
+    "deepdiff",
+    "nbmake==1.5.4",
+    "pytest-xdist==3.6.1"
 ]
 
 [tool.setuptools.dynamic]