diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index 395d7efaf67d0ca34256c561c8eee2e18fe957bf..c511a587b37176aa65575c20835f348a0467ebab 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -484,8 +484,8 @@
     "        # Prepare plotting arrays\n",
     "        step_timer.start()\n",
     "        if memory_cells == 1:\n",
-    "            fim_data[local_karabo_da] = data_corr[:plt_images, ...].copy()\n",
-    "            msk_data[local_karabo_da] = mask_corr[:plt_images, ...].copy()\n",
+    "            fim_data[local_karabo_da] = data_corr[:plt_images, 0, ...].copy()\n",
+    "            msk_data[local_karabo_da] = mask_corr[:plt_images, 0, ...].copy()\n",
     "        else:\n",
     "            fim_data[local_karabo_da] = data_corr[:plt_images, 1, ...].copy()\n",
     "            msk_data[local_karabo_da] = mask_corr[:plt_images, 1, ...].copy()\n",