diff --git a/crates/desktop/artcraft/src/services/grok/threads/grok_image_websocket_thread/grok_image_websocket_thread.rs b/crates/desktop/artcraft/src/services/grok/threads/grok_image_websocket_thread/grok_image_websocket_thread.rs
index c3d6e1dc43..fdf88e1f1e 100644
--- a/crates/desktop/artcraft/src/services/grok/threads/grok_image_websocket_thread/grok_image_websocket_thread.rs
+++ b/crates/desktop/artcraft/src/services/grok/threads/grok_image_websocket_thread/grok_image_websocket_thread.rs
@@ -319,7 +319,6 @@ async fn upload_images_to_storyteller(
       Some(&storyteller_creds),
       &task,
       &batch_token,
-
     ).await?;
 
     return Ok(())
@@ -332,7 +331,7 @@ async fn send_frontend_ui_update(
   maybe_creds: Option<&StorytellerCredentialSet>,
   task: &Task,
   batch_token: &BatchGenerationToken,
-) -> AnyhowResult<TextToImageGenerationCompleteEvent> {
+) -> AnyhowResult<()> {
 
   let result = list_batch_generated_redux_media_files(
     &app_env_configs.storyteller_host,
@@ -353,9 +352,13 @@ async fn send_frontend_ui_update(
       })
       .collect();
 
-  Ok(TextToImageGenerationCompleteEvent {
+  let event = TextToImageGenerationCompleteEvent {
     generated_images: media_files,
     maybe_frontend_subscriber_id: task.frontend_subscriber_id.clone(),
     maybe_frontend_subscriber_payload: task.frontend_subscriber_payload.clone(),
-  })
+  };
+
+  event.send_infallible(app);
+
+  Ok(())
 }
diff --git a/frontend/libs/model-list/src/lib/lists/ImageModels.ts b/frontend/libs/model-list/src/lib/lists/ImageModels.ts
index ce5a0a91d3..18e0f981ec 100644
--- a/frontend/libs/model-list/src/lib/lists/ImageModels.ts
+++ b/frontend/libs/model-list/src/lib/lists/ImageModels.ts
@@ -12,8 +12,8 @@ export const IMAGE_MODELS: ImageModel[] = [
     selectorName: "Grok",
     selectorDescription: "Fast af",
     selectorBadges: ["10 sec."],
-    maxGenerationCount: 4,
-    defaultGenerationCount: 4,
+    maxGenerationCount: 6,
+    defaultGenerationCount: 6,
     canUseImagePrompt: false,
     maxImagePromptCount: 6,
     progressBarTime: 10000,