diff --git a/app/chat/[id]/page.tsx b/app/chat/[id]/page.tsx
index a125141..f0d4034 100644
--- a/app/chat/[id]/page.tsx
+++ b/app/chat/[id]/page.tsx
@@ -1,9 +1,9 @@
 import { type Metadata } from 'next'
+import { auth } from '@clerk/nextjs'
 
 import { Chat } from '@/components/chat'
 import { getChat } from '@/app/actions'
 import { Header } from '@/components/header'
-import { auth } from '@clerk/nextjs'
 
 // export const runtime = 'edge'
 export const preferredRegion = 'home'
@@ -29,7 +29,7 @@ export default async function ChatPage({ params }: ChatPageProps) {
   const chat = await getChat(params.id, user?.id ?? '')
 
   return (
-    <div className="flex flex-col min-h-screen">
+    <div className="flex min-h-screen flex-col">
       {/* @ts-ignore */}
       <Header />
       <main className="flex-1 bg-muted/50">
diff --git a/components/chat-panel.tsx b/components/chat-panel.tsx
index 6bd0a40..3bd0d18 100644
--- a/components/chat-panel.tsx
+++ b/components/chat-panel.tsx
@@ -1,4 +1,4 @@
-import { type UseChatHelpers } from 'ai-connector'
+import { type UseChatHelpers } from 'ai-connector/react'
 
 import { Button } from '@/components/ui/button'
 import { ExternalLink } from '@/components/external-link'
diff --git a/components/empty-screen.tsx b/components/empty-screen.tsx
index e739b31..f47d1b0 100644
--- a/components/empty-screen.tsx
+++ b/components/empty-screen.tsx
@@ -1,8 +1,8 @@
-import { cn } from '@/lib/utils'
+import { UseChatHelpers } from 'ai-connector/react'
+
 import { Button } from '@/components/ui/button'
 import { ExternalLink } from '@/components/external-link'
 import { IconArrowRight } from '@/components/ui/icons'
-import { UseChatHelpers } from 'ai-connector'
 
 const exampleMessages = [
   {
diff --git a/components/prompt-form.tsx b/components/prompt-form.tsx
index 1fbc7b1..2dec66a 100644
--- a/components/prompt-form.tsx
+++ b/components/prompt-form.tsx
@@ -1,6 +1,7 @@
 import * as React from 'react'
 import Link from 'next/link'
 import Textarea from 'react-textarea-autosize'
+import { UseChatHelpers } from 'ai-connector/react'
 
 import { useEnterSubmit } from '@/lib/hooks/use-enter-submit'
 import { cn } from '@/lib/utils'
@@ -12,7 +13,6 @@ import {
   TooltipTrigger
 } from '@/components/ui/tooltip'
 import { IconArrowElbow, IconPlus } from '@/components/ui/icons'
-import { UseChatHelpers } from 'ai-connector'
 
 export interface PromptProps
   extends Pick<UseChatHelpers, 'input' | 'setInput'> {