diff --git a/apps/demo-nextjs-app-router/app/comfy/text_to_image/page.tsx b/apps/demo-nextjs-app-router/app/comfy/text_to_image/page.tsx
index 31d070e..0a2f3ed 100644
--- a/apps/demo-nextjs-app-router/app/comfy/text_to_image/page.tsx
+++ b/apps/demo-nextjs-app-router/app/comfy/text_to_image/page.tsx
@@ -84,7 +84,9 @@ export function Index() {
     const start = Date.now();
     try {
       const result: Result = await fal.subscribe('fal-ai/comfy-server', {
-        input: getWorkflow({}),
+        input: getWorkflow({
+          prompt: prompt,
+        }),
         pollInterval: 3000, // Default is 1000 (every 1s)
         logs: true,
         onQueueUpdate(update) {