diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index 395d7efaf67d0ca34256c561c8eee2e18fe957bf..4189dd347f4deb7bdd4c744a2e26e0b4894ceff4 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -320,7 +320,7 @@
     "        if memory_cells == 1:\n",
     "            g_data[index, ...] = g\n",
     "        else:\n",
-    "            g_data[index, ...] = g[1, ...]\n",
+    "            g_data[index, ...] = g[0, ...]\n",
     "    # Select memory cells\n",
     "    \n",
     "    # TODO: This needs to be revisited.\n",
@@ -411,7 +411,6 @@
     "            print(f\"\\t- WARNING: No image data for {ofile_name}: data shape is {dshape}\")\n",
     "            continue\n",
     "\n",
-    "        sensor_size = dshape[1:]\n",
     "        # load number of data available, including trains with empty data.\n",
     "        n_trains = len(seq_dc.train_ids)\n",
     "\n",
@@ -483,13 +482,9 @@
     "\n",
     "        # 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",
-    "        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",
-    "\n",
+    "        # TODO: Print out which cell is being selected for plotting\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",
     "        gim_data[local_karabo_da] = g_data\n",
     "        rim_data[local_karabo_da] = r_data\n",
     "        step_timer.done_step(f'Preparing plotting data of {plt_images} images:.')"