diff --git a/setup.py b/setup.py
index 0bda51c2..cc854f19 100644
--- a/setup.py
+++ b/setup.py
@@ -36,6 +36,7 @@
     install_requires=[
         "snakebids>=0.3.10",
         "snakemake>=6.5.2",
+        "batchgenerators==0.21",
         "nnunet-inference-on-cpu-and-gpu==1.6.6",
         "numpy>=1.20.2",
         "astropy",