diff --git a/notebooks/autogen/Autogendemo.ipynb b/notebooks/autogen/Autogendemo.ipynb
index 1b0896a..9d6d6cc 100644
--- a/notebooks/autogen/Autogendemo.ipynb
+++ b/notebooks/autogen/Autogendemo.ipynb
@@ -126,7 +126,7 @@
         }
       ],
       "source": [
-        "!pip install --quiet pyautogen litellm weaviate-client python-Levenshtein langchain langchain-openai langchain-community\n",
+        "!pip install --quiet ag2 litellm weaviate-client python-Levenshtein langchain langchain-openai langchain-community\n",
         "!pip install --quiet protobuf==3.20.0"
       ]
     },
diff --git a/notebooks/autogen/agentchat_RetrieveChat.ipynb b/notebooks/autogen/agentchat_RetrieveChat.ipynb
index 2ae502b..c66af9d 100644
--- a/notebooks/autogen/agentchat_RetrieveChat.ipynb
+++ b/notebooks/autogen/agentchat_RetrieveChat.ipynb
@@ -29,7 +29,7 @@
         "Some extra dependencies are needed for this notebook, which can be installed via pip:\n",
         "\n",
         "```bash\n",
-        "pip install pyautogen[retrievechat] flaml[automl]\n",
+        "pip install ag2[retrievechat] flaml[automl]\n",
         "```\n",
         "\n",
         "For more information, please refer to the [installation guide](/docs/installation/).\n",