diff --git a/apps/docs/components/Home.tsx b/apps/docs/components/Home.tsx
index 0417ac633..19460d0bc 100644
--- a/apps/docs/components/Home.tsx
+++ b/apps/docs/components/Home.tsx
@@ -5,8 +5,7 @@ import { Shadcn } from "@/components/shadcn/Shadcn";
 import { Badge } from "@/components/ui/badge";
 import { Button } from "@/components/ui/button";
 import { useChat } from "@ai-sdk/react";
-import { AssistantRuntimeProvider } from "@assistant-ui/react";
-import { useVercelUseChatRuntime } from "@assistant-ui/react-ai-sdk";
+import { AssistantRuntimeProvider, useEdgeRuntime } from "@assistant-ui/react";
 import Link from "next/link";
 import { useState } from "react";
 import { ChatGPT } from "./chatgpt/ChatGPT";
@@ -89,8 +88,7 @@ export type AssistantProps = {
 };
 
 const MyRuntimeProvider = ({ children }: { children: React.ReactNode }) => {
-  const chat = useChat();
-  const runtime = useVercelUseChatRuntime(chat);
+  const runtime = useEdgeRuntime({ api: "/api/chat" });
   return (
     <AssistantRuntimeProvider runtime={runtime}>
       {children}