diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 587a19993672dd3309813f29b27d5208ec2d63b7..296d39b1c3fc12a99d061cbfc71592e8e2575b35 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -336,10 +336,9 @@
    "source": [
     "# Before proceeding, verify the availability of trains\n",
     "# in AGIPD detector modules at INSTRUMENT group.\n",
-    "channels_to_correct = [instrument_src.format(m) for m in modules]\n",
-    "total_trains = len(dc.select(channels_to_correct, require_all=True).train_ids)\n",
-    "if total_trains == 0:\n",
-    "    latex_warning(f\"No trains available to correct in: {channels_to_correct}\")\n",
+    "dc_available_modules = dc.select(instrument_src.format(\"*\"), require_all=True)\n",
+    "if len(dc_available_modules.train_ids) == 0:\n",
+    "    latex_warning(f\"No trains available to correct in: {dc_available_modules}\")\n",
     "    sys.exit(0)"
    ]
   },