diff --git a/setup.py b/setup.py
index 6f4adafd..5f15139a 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
 
 
 install_requires = [
-    "pyautogen>=0.2.31",
+    "autogen-agentchat~=0.2",
     "graphviz>=0.20.1",
     "scikit-learn",
     "xgboost",