diff --git a/setup.py b/setup.py
index 57c0fa6..0c62032 100644
--- a/setup.py
+++ b/setup.py
@@ -7,13 +7,10 @@ setup(
     package_dir={"": "src"},
     install_requires=[
         "torch",
-        "aiia",
         "pillow",
         "torchvision",
         "scikit-learn",
-    ],
-    dependency_links=[
-        "git+https://gitea.fabelous.app/Machine-Learning/AIIA.git#egg=aiia"
+        "aiia @ git+https://gitea.fabelous.app/Machine-Learning/AIIA.git"
     ],
     python_requires=">=3.10",
 )
\ No newline at end of file