diff --git a/notebooks/NIRISS/NIRISS_WFSS_advanced/01_niriss_wfss_image2_image3.ipynb b/notebooks/NIRISS/NIRISS_WFSS_advanced/01_niriss_wfss_image2_image3.ipynb
index e49adbe82..91e230e3b 100644
--- a/notebooks/NIRISS/NIRISS_WFSS_advanced/01_niriss_wfss_image2_image3.ipynb
+++ b/notebooks/NIRISS/NIRISS_WFSS_advanced/01_niriss_wfss_image2_image3.ipynb
@@ -409,7 +409,6 @@
    "cell_type": "markdown",
    "id": "ca3bd322-7f67-43c4-8dae-dc47779462ee",
    "metadata": {
-    "editable": true,
     "slideshow": {
      "slide_type": ""
     },
@@ -1088,7 +1087,8 @@
    "source": [
     "# alternatively with a known source number\n",
     "# note that this number might be different for you depending on pipeline versions and parameters changed.\n",
-    "source = 118\n",
+    "# source = 109 # if you want to specify the number\n",
+    "source = cat['label'][nearest_id] # to match the one chosen in this notebook\n",
     "\n",
     "wh_source = np.where(np.array(cat['label'] == source))[0][0]\n",
     "\n",
@@ -1197,7 +1197,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.7"
+   "version": "3.9.7"
   }
  },
  "nbformat": 4,