diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index 92be5a793896982c66b9da225b3954ca00506396..49feebf94dbe5c85ed9afd26c861d957f1858e01 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -783,20 +783,11 @@
     "    out_folder / f for f in fnmatch.filter(corrected_files, f\"*{run}*S{first_seq:05d}*\")\n",
     "]\n",
     "\n",
-    "# TODO: replace with CALCAT value.\n",
-    "if \"1M\" in karabo_id:\n",
-    "    nmods = 2\n",
-    "elif \"4M\" in karabo_id:\n",
-    "    nmods = 8\n",
-    "else:  # 500K\n",
-    "    nmods = 1\n",
-    "\n",
     "with DataCollection.from_paths(seq_corrected_files) as corr_dc:\n",
     "    # Reading CORR data for plotting.\n",
     "    jf_corr = components.JUNGFRAU(\n",
     "        corr_dc,\n",
     "        detector_name=karabo_id,\n",
-    "        n_modules=nmods,\n",
     "    ).select_trains(np.s_[:plot_trains])\n",
     "    tid, jf_corr_data = next(iter(jf_corr.trains(require_all=True)))\n",
     "\n",
@@ -810,7 +801,7 @@
     "with RunDirectory(f\"{in_folder}/r{run:04d}/\", f\"*S{first_seq:05d}*\", _use_voview=False) as raw_dc:\n",
     "    # Reading RAW data for plotting.\n",
     "    jf_raw = components.JUNGFRAU(\n",
-    "        raw_dc, detector_name=karabo_id, n_modules=nmods, first_modno=jf_corr._modnos_start_at\n",
+    "        raw_dc, detector_name=karabo_id\n",
     "        ).select_trains(np.s_[:plot_trains])\n",
     "\n",
     "raw = jf_raw.get_array(\"data.adc\")[:, :, cell_idx_preview, ...]\n",