diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index f49c2c540b6eb28194e2658df639eca21000e1c9..fa33046e1fc4ecf0dc454931985dd0d61e53a402 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -210,7 +210,8 @@
     "    \"force_hg_if_below\",\n",
     "    \"force_mg_if_below\",\n",
     "    \"low_medium_gap\",\n",
-    "    \"melt_snow\"\n",
+    "    \"melt_snow\",\n",
+    "    \"rel_gain\"\n",
     "]"
    ]
   },
diff --git a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
index b92a901a9c82422651601445255b6ab2df4b0d96..c081527a552c1356823162f228bae82e571f6117 100644
--- a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
+++ b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
@@ -339,7 +339,6 @@
     "for module_index, k_da in zip(modules, karabo_da):\n",
     "    qm = tools.module_index_to_qm(module_index)\n",
     "    if qm in mapped_files and not mapped_files[qm].empty():\n",
-    "        device = getattr(getattr(Detectors, dinstance), qm)\n",
     "        # TODO: make map_modules_from_folder just return list(s)\n",
     "        qm_files = [Path(mapped_files[qm].get()) for _ in range(mapped_files[qm].qsize())]\n",
     "    else:\n",