diff --git a/setup.py b/setup.py
index eab33fc..110c2ae 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
     python_requires=">=3.8",
     install_requires=[
         "torch<=2.2.1",
-        "nnunetv2",
+        "nnunetv2==2.5.1",
         "argparse",
         "huggingface_hub"
     ],