diff --git a/app/(chat)/page.tsx b/app/(chat)/page.tsx
index abb7e5fc20..b134d7c177 100644
--- a/app/(chat)/page.tsx
+++ b/app/(chat)/page.tsx
@@ -26,30 +26,14 @@ async function NewChatPage() {
 
   const cookieStore = await cookies();
   const modelIdFromCookie = cookieStore.get("chat-model");
-
-  if (!modelIdFromCookie) {
-    return (
-      <>
-        <Chat
-          autoResume={false}
-          id={id}
-          initialChatModel={DEFAULT_CHAT_MODEL}
-          initialMessages={[]}
-          initialVisibilityType="private"
-          isReadonly={false}
-          key={id}
-        />
-        <DataStreamHandler />
-      </>
-    );
-  }
+  const initialChatModel = modelIdFromCookie?.value || DEFAULT_CHAT_MODEL;
 
   return (
     <>
       <Chat
         autoResume={false}
         id={id}
-        initialChatModel={modelIdFromCookie.value}
+        initialChatModel={initialChatModel}
         initialMessages={[]}
         initialVisibilityType="private"
         isReadonly={false}