diff --git a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
index 520666c8579b6e3e1954b840d1b5fedf9f53d7d6..f7d895b826a7f3715fe9fbe1732e8e5b125c70d0 100644
--- a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
+++ b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
@@ -263,7 +263,11 @@
     "instrument_src_mod = instrument_src.format(modules[0])\n",
     "\n",
     "agipd_ctrl_dark = AgipdCtrlRuns(\n",
-    "    raw_folder=in_folder, runs=list(runs_dict.keys()))\n",
+    "    raw_folder=in_folder,\n",
+    "    runs=[r[\"number\"] for r in runs_dict.values()],\n",
+    "    image_src=instrument_src_mod,\n",
+    "    ctrl_src=ctrl_src,\n",
+    ")\n",
     "if mem_cells == 0:\n",
     "    mem_cells = agipd_ctrl_dark.get_memory_cells()\n",
     "\n",