diff --git a/setup.py b/setup.py
index b306346..4fc7682 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,12 @@
     url="https://github.com/CBICA/DLICV/",
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.9",
-    install_requires=["torch<=2.2.1", "nnunetv2<=2.2.1", "argparse", "huggingface_hub"],
+    install_requires=[
+        "torch<=2.2.1",
+        "nnunetv==2.5.1",
+        "argparse",
+        "huggingface_hub"
+    ],
     entry_points={"console_scripts": ["DLICV = DLICV.__main__:main"]},
     classifiers=[
         "Intended Audience :: Developers",