diff --git a/app/components/chat.tsx b/app/components/chat.tsx
index ed51d926f..5669cc9a3 100644
--- a/app/components/chat.tsx
+++ b/app/components/chat.tsx
@@ -2071,6 +2071,6 @@ function _Chat() {
 
 export function Chat() {
   const chatStore = useChatStore();
-  const sessionIndex = chatStore.currentSessionIndex;
-  return <_Chat key={sessionIndex}></_Chat>;
+  const session = chatStore.currentSession();
+  return <_Chat key={session.id}></_Chat>;
 }
diff --git a/app/utils.ts b/app/utils.ts
index 2dd80b8a3..1c359ef95 100644
--- a/app/utils.ts
+++ b/app/utils.ts
@@ -257,11 +257,11 @@ export function isVisionModel(model: string) {
   const excludeKeywords = ["claude-3-5-haiku-20241022"];
   const visionKeywords = [
     "vision",
-    "claude-3",
-    "gemini-1.5-pro",
-    "gemini-1.5-flash",
     "gpt-4o",
-    "gpt-4o-mini",
+    "claude-3",
+    "gemini-1.5",
+    "qwen-vl",
+    "qwen2-vl",
   ];
   const isGpt4Turbo =
     model.includes("gpt-4-turbo") && !model.includes("preview");