diff --git a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
index bbb32e928c2784a85ea2c1b28b58a1c39cb8388f..f05b3cf1bec5a565338c512c42518b643c220b14 100644
--- a/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
+++ b/notebooks/AGIPD/AGIPD_Retrieve_Constants_Precorrection.ipynb
@@ -176,12 +176,16 @@
    "outputs": [],
    "source": [
     "run_dc = RunDirectory(in_folder / f\"r{run:04d}\")\n",
-    "\n",
     "instrument_src = instrument_source_template.format(karabo_id, receiver_template)\n",
     "\n",
-    "instr_dc = run_dc.select(instrument_src.format(\"*\"), require_all=True)\n",
+    "instr_dc = run_dc.select(instrument_src.format(\"*\"))\n",
+    "\n",
+    "for m in modules:\n",
+    "    # Remove empty sources from `instr_dc`\n",
+    "    if instr_dc[instrument_src.format(m), 'image.data'].shape[0] == 0:\n",
+    "        instr_dc = instr_dc.deselect(instrument_src.format(m))\n",
     "\n",
-    "if not instr_dc.train_ids:\n",
+    "if not instr_dc.all_sources:\n",
     "    raise ValueError(f\"No images found for {in_folder / f'r{run:04d}'}\")"
    ]
   },