diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 2208d4ddbbe5f6ac286dfed86dc0eac1ea9278dc..9a2da6a73ab0a5cf6a0f62143cfaae6fcd047d51 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -339,7 +339,7 @@
     "for m in modules:\n",
     "    try:\n",
     "        # Attempt to select the module. If no trains are available, ValueError might be raised\n",
-    "        if len(dc.select(instrument_src.format(m), require_all=True).train_ids) > 0:\n",
+    "        if len(dc[instrument_src.format(m), 'image.data'].drop_empty_trains().train_ids) > 0:\n",
     "            train_available = True\n",
     "            break  # Found a module with available trains.\n",
     "    except ValueError as e:\n",