diff --git a/template/main.py b/template/main.py
index 3fec05726..4977c3799 100644
--- a/template/main.py
+++ b/template/main.py
@@ -12,14 +12,14 @@
     BedrockModels,
     GeminiModels,
     OpenAIModels,
-    UiPathChatBedrock,
+    UiPathChatBedrockConverse,
     UiPathChatOpenAI,
     UiPathChatVertex,
 )
 
 # Choose your LLM provider by uncommenting one of the following:
-llm = UiPathChatVertex(model_name=GeminiModels.gemini_2_5_flash)
-# llm = UiPathChatBedrock(model_name=BedrockModels.anthropic_claude_haiku_4_5)
+llm = UiPathChatBedrockConverse(model_name=BedrockModels.anthropic_claude_haiku_4_5)
+# llm = UiPathChatVertex(model_name=GeminiModels.gemini_2_5_flash)
 # llm = UiPathChatOpenAI(model_name=OpenAIModels.gpt_4_1_mini_2025_04_14)
 
 SYSTEM_PROMPT = (
diff --git a/testcases/template-agent/expected_traces.json b/testcases/template-agent/expected_traces.json
index 1b3b3c231..6535430c7 100644
--- a/testcases/template-agent/expected_traces.json
+++ b/testcases/template-agent/expected_traces.json
@@ -9,7 +9,7 @@
     },
     {
       "name": [
-        "UiPathChatBedrock",
+        "UiPathChatBedrockConverse",
         "UiPathChatOpenAI",
         "UiPathChatVertex"
       ],