diff --git a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
index 96d7a18a8e2c13068a13954c88e0efb97921d7a7..0679cfd1ceac81be21316da1223f4eebdfb35016 100644
--- a/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
+++ b/notebooks/AGIPD/Characterize_AGIPD_Gain_Darks_NBC.ipynb
@@ -586,19 +586,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",