diff --git a/dress/datasetgeneration/json_schema.py b/dress/datasetgeneration/json_schema.py
index 0ea68cd..05c87e0 100644
--- a/dress/datasetgeneration/json_schema.py
+++ b/dress/datasetgeneration/json_schema.py
@@ -13,6 +13,7 @@
                 "outbasename": {"type": ["null", "string"]},
                 "seed": {"type": "integer"},
                 "model": {"type": "string"},
+                "batch_size": {"type": "integer"},
                 "model_scoring_metric": {"type": "string"},
                 "pangolin_mode": {"type": "string"},
                 "pangolin_tissue": {"type": ["null", "string"]},