diff --git a/.github/workflows/check_notebooks.yml b/.github/workflows/check_notebooks.yml
index 969a5cb..ac9baaa 100644
--- a/.github/workflows/check_notebooks.yml
+++ b/.github/workflows/check_notebooks.yml
@@ -56,6 +56,5 @@
           - name: Run notebooks
             run: |
               find . -name "*.ipynb" -exec sed -Ei "s/num_epochs=.?.?/num_epochs=1/g" {} +
-              find . -name "*.ipynb" -exec sed -i "s/PatchDataset(patch_df,/PatchDataset(patch_df[:1],/g" {} +
-              find . -name "*.ipynb" -exec sed -i "s/PatchContextDataset(patch_df,/PatchDataset(patch_df[:1],/g" {} +
-              find . -name "*.ipynb" -exec bash -c 'papermill --cwd $(dirname {}) {} {}' \;
\ No newline at end of file
+              find . -name "*.ipynb" -exec sed -i "s/annotator.annotate/#annotator.annotate/g" {} +
+              find . -name "*.ipynb" -exec bash -c 'papermill --cwd $(dirname {}) --log-output {} {}' \;
\ No newline at end of file