diff --git a/python/packages/autogen-studio/frontend/src/components/types/guards.ts b/python/packages/autogen-studio/frontend/src/components/types/guards.ts
index 63e6a5a1c706..726a40a35188 100644
--- a/python/packages/autogen-studio/frontend/src/components/types/guards.ts
+++ b/python/packages/autogen-studio/frontend/src/components/types/guards.ts
@@ -34,8 +34,7 @@ const PROVIDERS = {
 
   // Models
   OPENAI: "autogen_ext.models.openai.OpenAIChatCompletionClient",
-  AZURE_OPENAI:
-    "autogen_ext.models.azure_openai.AzureOpenAIChatCompletionClient",
+  AZURE_OPENAI: "autogen_ext.models.openai.AzureOpenAIChatCompletionClient",
 
   // Tools
   FUNCTION_TOOL: "autogen_core.tools.FunctionTool",