diff --git a/setup.py b/setup.py
index 1d137444..f2800aae 100644
--- a/setup.py
+++ b/setup.py
@@ -72,6 +72,6 @@ setuptools.setup(
         'dill~=0.3',
         'xgboost~=1.3',
         # for e2e dialogues:
-        'dialogues @ git+https://github.com/Mehrad0711/dialogues@main',
+        'dialogues @ git+https://github.com/Mehrad0711/dialogues@wip/dev_1',
     ],
 )