diff --git a/ansible_ai_connect_chatbot/src/App.test.tsx b/ansible_ai_connect_chatbot/src/App.test.tsx
index 17744e0ab..40dada064 100644
--- a/ansible_ai_connect_chatbot/src/App.test.tsx
+++ b/ansible_ai_connect_chatbot/src/App.test.tsx
@@ -402,7 +402,10 @@ test("Debug mode test", async () => {
   await expect
     .element(view.getByRole("menuitem", { name: "granite3-8b" }))
     .toBeTruthy();
-  await view.getByRole("menuitem", { name: "granite3-8b" }).click();
+  await expect
+    .element(view.getByRole("menuitem", { name: "granite31-8b" }))
+    .toBeTruthy();
+  await view.getByRole("menuitem", { name: "granite31-8b" }).click();
 
   await sendMessage("Hello");
   await expect
diff --git a/ansible_ai_connect_chatbot/src/useChatbot/useChatbot.ts b/ansible_ai_connect_chatbot/src/useChatbot/useChatbot.ts
index 4ff8ce3ae..d75199105 100644
--- a/ansible_ai_connect_chatbot/src/useChatbot/useChatbot.ts
+++ b/ansible_ai_connect_chatbot/src/useChatbot/useChatbot.ts
@@ -25,6 +25,7 @@ const botName =
 
 export const modelsSupported: LLMModel[] = [
   { model: "granite3-8b", provider: "my_rhoai_g3" },
+  { model: "granite31-8b", provider: "my_rhoai_g31" },
 ];
 
 export const readCookie = (name: string): string | null => {