diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 147757da0d354aec3c12ee9a6dcc0ee84c697bfa..34038d771f0da0b1afb4991150e5aec85f8a1cb9 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -21,6 +21,7 @@
     "out_folder = \"/gpfs/exfel/data/scratch/ahmedk/test/hibef_agipd2\"  # the folder to output to, required\n",
     "sequences = [-1] # sequences to correct, set to -1 for all, range allowed\n",
     "modules = [-1] # modules to correct, set to -1 for all, range allowed\n",
+    "train_ids = [-1] # train IDs to correct, set to -1 for all, range allowed\n",
     "run = 155 # runs to process, required\n",
     "\n",
     "karabo_id = \"HED_DET_AGIPD500K2G\" # karabo karabo_id\n",
@@ -390,8 +391,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Obtain trains to process if using a pulse picker device.\n",
     "if use_ppu_device:\n",
+    "    # Obtain trains to process if using a pulse picker device.\n",
     "    run = RunDirectory(in_folder / f'r{run:04d}')\n",
     "\n",
     "    # Will throw an uncaught exception if the device is wrong.\n",
@@ -403,7 +404,11 @@
     "\n",
     "    print(f'PPU device {use_ppu_device} triggered for {len(train_ids)} trains')\n",
     "\n",
+    "elif train_ids[0] != [-1]:\n",
+    "    # Specific trains passed by parameter, convert to ndarray.\n",
+    "    train_ids = np.array(train_ids)\n",
     "else:\n",
+    "    # Process all trains.\n",
     "    train_ids = None"
    ]
   },