diff --git a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
index 9201b66fd681296ea84b759a5483bd6dae0c1426..3397ba745fd4cf5db416ca4742908443cf744b46 100644
--- a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
+++ b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
@@ -504,13 +504,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Add a badpixel due to bad gain separation\n",
+    "# Add bad pixels due to bad gain separation\n",
     "if not fixed_gain_mode:\n",
-    "    for g in range(2):\n",
-    "        # Bad pixels during bad gain separation.\n",
-    "        # Fraction of pixels in the module with separation lower than \"thresholds_gain_sigma\".\n",
-    "        bad_sep = (gain_g[qm][..., g+1] - gain_g[qm][..., g]) / np.sqrt(gainstd_g[qm][..., g+1]**2 + gainstd_g[qm][..., g]**2)\n",
-    "        badpix_g[qm][...,g+1][(bad_sep)<thresholds_gain_sigma]|= BadPixels.GAIN_THRESHOLDING_ERROR"
+    "    for qm in gain_g.keys():\n",
+    "        for g in range(2):\n",
+    "            # Bad pixels during bad gain separation.\n",
+    "            # Fraction of pixels in the module with separation lower than \"thresholds_gain_sigma\".\n",
+    "            bad_sep = (gain_g[qm][..., g+1] - gain_g[qm][..., g]) / \\\n",
+    "                np.sqrt(gainstd_g[qm][..., g+1]**2 + gainstd_g[qm][..., g]**2)\n",
+    "            badpix_g[qm][...,g+1][bad_sep<thresholds_gain_sigma] |= \\\n",
+    "                BadPixels.GAIN_THRESHOLDING_ERROR"
    ]
   },
   {