diff --git a/.github/workflows/notebooks.yaml b/.github/workflows/notebooks.yaml
index fd3d416..877ee26 100644
--- a/.github/workflows/notebooks.yaml
+++ b/.github/workflows/notebooks.yaml
@@ -17,6 +17,8 @@ jobs:
           python -c "import torch; print('Is CUDA available:', torch.cuda.is_available())"
 
       - name: Import flow_matching
+        shell: bash -l {0}
+        working-directory: examples
         run: |
           python -c "import flow_matching; print('flow_matching module imported successfully')"