diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index 2ae33c6ea3e231508daea839dd293d89f956f083..2cbf2883903625f971734edb684e66e5c8011853 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -212,9 +212,6 @@
     "        gain_mode=gm,\n",
     "        client=rest_cfg.calibration_client(),\n",
     "    )\n",
-    "    da_to_pdu = {}\n",
-    "    for mod_info in jf_cal.physical_detector_units.values():\n",
-    "        da_to_pdu[mod_info[\"karabo_da\"]] = mod_info[\"physical_name\"]\n",
     "\n",
     "    constant_names = [\"Offset10Hz\", \"BadPixelsDark10Hz\"]\n",
     "    if relative_gain:\n",
@@ -325,7 +322,12 @@
     "\n",
     "# load constants arrays after storing fragment YAML file\n",
     "# and validating constants availability.\n",
-    "const_data = jf_cal.ndarray_map(metadata=jf_metadata)"
+    "const_data = jf_cal.ndarray_map(metadata=jf_metadata)\n",
+    "\n",
+    "# For plotting\n",
+    "da_to_pdu = {}\n",
+    "for mod_info in jf_cal.physical_detector_units.values():\n",
+    "    da_to_pdu[mod_info[\"karabo_da\"]] = mod_info[\"physical_name\"]"
    ]
   },
   {