diff --git a/app/page.tsx b/app/page.tsx
index 6e67e49..e24cde8 100644
--- a/app/page.tsx
+++ b/app/page.tsx
@@ -4,35 +4,28 @@ import { DrawIoEmbed } from "react-drawio";
 import ChatPanel from "@/components/chat-panel";
 import { DiagramProvider, useDiagram } from "@/contexts/diagram-context";
 
-// Internal layout component
-function DiagramPageLayout({ children }: { children: React.ReactNode }) {
+export default function Home() {
     const { drawioRef, handleDiagramExport } = useDiagram();
 
-    return (
-        <div className="flex h-screen bg-gray-100">
-            <div className="w-2/3 p-1 h-full relative">
-                <DrawIoEmbed
-                    ref={drawioRef}
-                    onExport={handleDiagramExport}
-                    urlParameters={{
-                        spin: true,
-                        libraries: false,
-                        saveAndExit: false,
-                        noExitBtn: true,
-                    }}
-                />
-            </div>
-            <div className="w-1/3 h-full p-1">{children}</div>
-        </div>
-    );
-}
-
-export default function Home() {
     return (
         <DiagramProvider>
-            <DiagramPageLayout>
-                <ChatPanel />
-            </DiagramPageLayout>
+            <div className="flex h-screen bg-gray-100">
+                <div className="w-2/3 p-1 h-full relative">
+                    <DrawIoEmbed
+                        ref={drawioRef}
+                        onExport={handleDiagramExport}
+                        urlParameters={{
+                            spin: true,
+                            libraries: false,
+                            saveAndExit: false,
+                            noExitBtn: true,
+                        }}
+                    />
+                </div>
+                <div className="w-1/3 h-full p-1">
+                    <ChatPanel />
+                </div>
+            </div>
         </DiagramProvider>
     );
 }