diff --git a/src/components/ai-optimized-editor.tsx b/src/components/ai-optimized-editor.tsx
index e2eeac2..40a2433 100644
--- a/src/components/ai-optimized-editor.tsx
+++ b/src/components/ai-optimized-editor.tsx
@@ -130,6 +130,7 @@ export function AIProblemEditor({
           modified={optimizedCode}
           language="typescript"
           theme="vs-dark"
+          className="h-full w-full"
           options={{
             readOnly: true,
             minimap: { enabled: false }