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