diff --git a/setup.py b/setup.py
index 1ceda5c04..ada5fcb28 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,7 @@ def parse_requirements():
     if "torch==2.1.0" in _install_requires:
         _install_requires.pop(_install_requires.index("xformers>=0.0.22"))
         _install_requires.append(
-            "git+https://github.com/facebookresearch/xformers.git@main#egg=xformers"
+            "xformers @ git+https://github.com/facebookresearch/xformers.git@main"
         )
 
     return _install_requires, _dependency_links