diff --git a/tests/test_reference_runs/callab_tests.py b/tests/test_reference_runs/callab_tests.py
index 9ba705b2c28d3f84b9a0b5fc4859fc3af497cd73..3ddfbe30c5d0f0fc1dc5ba71652a38ee7e53ec36 100644
--- a/tests/test_reference_runs/callab_tests.py
+++ b/tests/test_reference_runs/callab_tests.py
@@ -29,6 +29,7 @@ automated_test_config = {
             "run-med": "9012",  # Original run: "92"
             "run-low": "9013",  # Original run "93"
             "karabo-id-control": "SPB_IRU_AGIPD1M1",
+            "max-trains": 250,
             "karabo-id": "SPB_DET_AGIPD1M-1",
             "ctrl-source-template": "{}/MDL/FPGA_COMP",
         },
@@ -65,6 +66,7 @@ automated_test_config = {
             "run-low": "9017",  # Original run: "218",
             "karabo-id-control": "SPB_IRU_AGIPD1M1",
             "karabo-id": "SPB_DET_AGIPD1M-1",
+            "max-trains": 250,
             "ctrl-source-template": "{}/MDL/FPGA_COMP",
         },
         "reference-folder": "{}/{}/{}",
@@ -143,6 +145,7 @@ automated_test_config = {
             "run-low": "9021",  # Original run: "6"
             "karabo-id-control": "MID_EXP_AGIPD1M1",
             "karabo-id": "MID_DET_AGIPD1M-1",
+            "max-trains": 250,
             "ctrl-source-template": "{}/MDL/FPGA_COMP",
         },
         "reference-folder": "{}/{}/{}",
@@ -176,6 +179,7 @@ automated_test_config = {
             "run-high": "9023", # Original run: "25",
             "run-med": "9024",  # Original run: "26",
             "run-low": "9025",  # Original run: "27",
+            "max-trains": 250,
             "karabo-id-control": "HED_EXP_AGIPD500K2G",
             "karabo-id": "HED_DET_AGIPD500K2G",
             "ctrl-source-template": "{}/MDL/FPGA_COMP",