diff --git a/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/agents.ipynb b/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/agents.ipynb
index 9c01c45b1..05aef4bfd 100644
--- a/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/agents.ipynb
+++ b/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/agents.ipynb
@@ -312,7 +312,8 @@
     "\n",
     "from autogen_agentchat.agents import BaseChatAgent\n",
     "from autogen_agentchat.base import Response\n",
-    "from autogen_agentchat.messages import AgentMessage, ChatMessage\n",
+    "from autogen_agentchat.messages import AgentMessage, ChatMessage, TextMessage\n",
+    "from autogen_core.base import CancellationToken\n",
     "\n",
     "\n",
     "class CountDownAgent(BaseChatAgent):\n",
@@ -437,7 +438,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.5"
+   "version": "3.12.5"
   }
  },
  "nbformat": 4,