diff --git a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
index bb9cf91e0fd39a9f29c4d78a821dce6ad42b91fb..ada299a6fc235f7c7d976b112001ee72a57eeca7 100644
--- a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
+++ b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
@@ -593,19 +593,8 @@
     ")\n",
     "karabo_da_to_pdu = {d[\"karabo_da\"]: d[\"pdu_physical_name\"] for d in all_pdus}\n",
     "for qm_attr in qm_dict.values():\n",
-    "    qm_attr[\"db_module\"] = karabo_da_to_pdu[qm_attr[\"karabo_da\"]]\n",
-    "for pdu_dict in all_pdus:\n",
-    "    this_karabo_da = pdu_dict[\"karabo_da\"]\n",
-    "    if this_karabo_da in qm_dict:\n",
-    "        qm_dict[this_karabo_da][\"db_module\"] = pdu_dict[\"pdu_physical_name\"]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
+    "    qm_attr[\"db_module\"] = karabo_da_to_pdu[qm_attr[\"karabo_da\"]]    \n",
+    "\n",
     "with open(f\"{out_folder}/module_mapping.yml\",\"w\") as fd:\n",
     "    yaml.safe_dump({\n",
     "        \"module_mapping\": {\n",