diff --git a/notebooks/Jungfrau/Jungfrau_Create_Fit_Spectra_Histos_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Create_Fit_Spectra_Histos_NBC.ipynb
index c6025ee092757085593441aade89e4b4405a4e24..f3baf2912eb575adf35be4b5c881466638fd89ef 100644
--- a/notebooks/Jungfrau/Jungfrau_Create_Fit_Spectra_Histos_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Create_Fit_Spectra_Histos_NBC.ipynb
@@ -343,7 +343,7 @@
     "    # Jungfrau gains 0[00], 1[01], 3[11]\n",
     "    # Change low gain to 2 for indexing purposes.\n",
     "    g[g==3] = 2\n",
-    "\n",
+    "    m = 0\n",
     "    # Select memory cells\n",
     "    if memory_cells > 1:\n",
     "        \"\"\"\n",
@@ -360,7 +360,7 @@
     "        m[m==255] = 0\n",
     "        offset_map_cell = offset_map[:, m, ...]  # [16 + empty cell, x, y]\n",
     "    else:\n",
-    "        offset_map_cell = offset_map[:, 0, ...]\n",
+    "        offset_map_cell = offset_map[:, m, ...]\n",
     "\n",
     "    # Offset correction\n",
     "    offset = np.choose(g, offset_map_cell)\n",