diff --git a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
index 1e4d4ded89f7b326f19e58a5c018f9e161024e24..a14dea206177ba49c37437b3ec1dfa744d0811a6 100644
--- a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
+++ b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
@@ -565,7 +565,8 @@
     "        condition = Conditions.Dark.AGIPD(memory_cells=max_cells,\n",
     "                                          bias_voltage=bias_voltage,\n",
     "                                          acquisition_rate=acq_rate,\n",
-    "                                          gain_setting=gain_setting)\n",
+    "                                          gain_setting=gain_setting,\n",
+    "                                          gain_mode=(True if fixed_gain_mode else None))\n",
     "\n",
     "        # This should be used in case of running notebook\n",
     "        # by a different method other than myMDC which already\n",
@@ -614,7 +615,8 @@
     "        condition = Conditions.Dark.AGIPD(memory_cells=max_cells,\n",
     "                                          bias_voltage=bias_voltage,\n",
     "                                          acquisition_rate=acq_rate,\n",
-    "                                          gain_setting=gain_setting)\n",
+    "                                          gain_setting=gain_setting,\n",
+    "                                          gain_mode=(True if fixed_gain_mode else None))\n",
     "        if db_output:\n",
     "            md = send_to_db(db_module, karabo_id, dconst, condition, file_loc,\n",
     "                            report, cal_db_interface, creation_time=creation_time,\n",