diff --git a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
index afcbd7c43bc3952fec3e6965f5d7f55c1fe7ab6a..95bb91ef5f09434db26126aeed0bd6772059d668 100644
--- a/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
+++ b/notebooks/Jungfrau/Jungfrau_Gain_Correct_and_Verify_NBC.ipynb
@@ -398,6 +398,18 @@
     "    else:\n",
     "        gain_corr[index, ...] = g\n",
     "\n",
+    "    # Jungfrau gains 0[00], 1[01], 3[11]\n",
+    "    # Change low gain to 2 for indexing purposes.\n",
+    "    g[g==3] = 2\n",
+    "\n",
+    "    if replace_wrong_gain_value > -1:\n",
+    "        g = replace_pixels_gain_values(\n",
+    "            g,\n",
+    "            replace_wrong_gain_value,\n",
+    "            wrong_gain_pixels,\n",
+    "            local_karabo_da,\n",
+    "        )\n",
+    "\n",
     "    # Select memory cells\n",
     "    if memory_cells > 1:\n",
     "        \"\"\"\n",
@@ -418,20 +430,6 @@
     "        offset_map_cell = offset_map\n",
     "        mask_cell = mask\n",
     "\n",
-    "    # Jungfrau gains 0[00], 1[01], 3[11]\n",
-    "    # Change low gain to 2 for indexing purposes.\n",
-    "    g[g==3] = 2\n",
-    "\n",
-    "    if replace_wrong_gain_value > -1:\n",
-    "        g = replace_pixels_gain_values(\n",
-    "            g,\n",
-    "            replace_wrong_gain_value,\n",
-    "            wrong_gain_pixels,\n",
-    "            local_karabo_da,\n",
-    "            mask_cell,\n",
-    "            memory_cells,\n",
-    "        )\n",
-    "\n",
     "    # Offset correction\n",
     "    offset = np.choose(g, np.moveaxis(offset_map_cell, -1, 0))\n",
     "\n",