diff --git a/.github/workflows/notebooks.yaml b/.github/workflows/notebooks.yaml
index 483030d..6388e36 100644
--- a/.github/workflows/notebooks.yaml
+++ b/.github/workflows/notebooks.yaml
@@ -21,7 +21,14 @@ jobs:
         env:
           PYTHONPATH: $PYTHONPATH:$PWD
         run: |
-          set -e
+          set +e
+          
+          cd examples/
+          python -m flow_matching "import flow_matching; print('flow_matching module imported successfully')"
+          python -c "import flow_matching; print('flow_matching module imported successfully')"
+          python -c "import sys; print(sys.path)"
+          cd ..
+          
           for file in examples/*.ipynb; do
-            jupyter nbconvert --to notebook --execute --ExecutePreprocessor.timeout=300 "$file"
+            PYTHONPATH=$PYTHONPATH jupyter nbconvert --to notebook --execute --ExecutePreprocessor.timeout=300 "$file"
           done
\ No newline at end of file