diff --git a/frontend/libs/components/promptbox/src/lib/PromptBoxImage.tsx b/frontend/libs/components/promptbox/src/lib/PromptBoxImage.tsx
index 4e7ac4ce04..d009b4d144 100644
--- a/frontend/libs/components/promptbox/src/lib/PromptBoxImage.tsx
+++ b/frontend/libs/components/promptbox/src/lib/PromptBoxImage.tsx
@@ -268,7 +268,7 @@ export const PromptBoxImage = ({
       prompt: prompt,
       model: modelInfo,
       aspect_ratio: aspectRatio,
-      //generation_count: generationCount, TODO: Backend doesn't support generation count yet?
+      number_images: generationCount,
     });
 
     console.log("PromptBoxImage - generateResponse", generateResponse);