diff --git a/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/models.ipynb b/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/models.ipynb
index a0b9d1cb4..4700897e7 100644
--- a/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/models.ipynb
+++ b/python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/tutorial/models.ipynb
@@ -165,6 +165,7 @@
     "from autogen_ext.auth.azure import AzureTokenProvider\n",
     "from autogen_ext.models.openai import AzureOpenAIChatCompletionClient\n",
     "from azure.identity import DefaultAzureCredential\n",
+    "from autogen_core.models import UserMessage\n",
     "\n",
     "# Create the token provider\n",
     "token_provider = AzureTokenProvider(\n",