diff --git a/notebooks/run_Cellpose-SAM.ipynb b/notebooks/run_Cellpose-SAM.ipynb
index 5f2232af..543d186f 100644
--- a/notebooks/run_Cellpose-SAM.ipynb
+++ b/notebooks/run_Cellpose-SAM.ipynb
@@ -358,8 +358,8 @@
       "source": [
         "for i in trange(len(files)):\n",
         "    f = files[i]\n",
-        "    masks0 = io.imsave(dir / (f.name + \"_masks\" + masks_ext))\n",
-        "    io.save_rois(masks0, f)"
+        "    io.imsave(dir / (f.name + \"_masks\" + masks_ext), masks[i])\n",
+        "    io.save_rois(masks[i], f)"
       ]
     }
   ],