diff --git a/gpt4all_api/api.py b/gpt4all_api/api.py
index 945ba8bb..68ea105a 100644
--- a/gpt4all_api/api.py
+++ b/gpt4all_api/api.py
@@ -253,7 +253,10 @@ class ModelProcess:
                             if self.personality.processor_cfg is not None:
                                 if "custom_workflow" in self.personality.processor_cfg:
                                     if self.personality.processor_cfg["custom_workflow"]:
-                                        self.personality.processor.run_workflow(self._generate, command[1], command[0])
+                                        output = self.personality.processor.run_workflow(self._generate, command[1], command[0])
+                                        self._callback(output)
+                                        self.is_generating.value = 0
+                                        continue
 
                         self._generate(command[0], self._callback)
                         while not self.generation_queue.empty():