diff --git a/app/api/generate/route.ts b/app/api/generate/route.ts
index 53ccc59..c034158 100644
--- a/app/api/generate/route.ts
+++ b/app/api/generate/route.ts
@@ -26,10 +26,10 @@ export async function POST(request: NextRequest) {
         const generatedFiles = await generateApp(description);
 
         // // Parse through schema before passing to repair
-        // const validatedFiles = benchifyFileSchema.parse(generatedFiles);
+        const validatedFiles = benchifyFileSchema.parse(generatedFiles);
 
         // // Repair the generated code using Benchify's API
-        // const repairedFiles = await repairCode(validatedFiles);
+        const repairedFiles = await repairCode(validatedFiles);
 
         const { sbxId, template, url } = await createSandbox({ files: generatedFiles });
 
@@ -38,7 +38,7 @@ export async function POST(request: NextRequest) {
         // Return the results to the client
         return NextResponse.json({
             originalFiles: generatedFiles,
-            // repairedFiles: repairedFiles,
+            repairedFiles: repairedFiles,
             // buildOutput: '',  // We don't get build output from Benchify in our current setup
             previewUrl: url,
         });
diff --git a/app/page.tsx b/app/page.tsx
index eaa67a5..542097f 100644
--- a/app/page.tsx
+++ b/app/page.tsx
@@ -1,13 +1,18 @@
 // app/page.tsx
 'use client';
 
-import { useState } from 'react';
+import { useEffect, useState } from 'react';
 import { PromptForm } from '@/components/ui-builder/prompt-form';
 import { Card, CardContent } from '@/components/ui/card';
-
 export default function Home() {
   const [result, setResult] = useState<any>(null);
 
+  useEffect(() => {
+    if (result) {
+      console.log(result);
+    }
+  }, [result]);
+
   return (
     <main className="min-h-screen flex flex-col items-center justify-center bg-background">
       <div className="w-full max-w-3xl mx-auto">