diff --git a/requirements.txt b/requirements.txt
index 7a9cbca..d06fb2e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 torch==2.2.1
-nnunetv2==2.5.1
+nnunetv2==2.2.1
 argparse
 huggingface_hub
diff --git a/setup.py b/setup.py
index 4fc7682..8b763a5 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
     python_requires=">=3.9",
     install_requires=[
         "torch<=2.2.1",
-        "nnunetv==2.5.1",
+        "nnunetv<=2.2.1",
         "argparse",
         "huggingface_hub"
     ],