diff --git a/api/__init__.py b/api/__init__.py
index ae974fa0..463572d9 100644
--- a/api/__init__.py
+++ b/api/__init__.py
@@ -488,15 +488,12 @@ class LoLLMsAPPI():
                                         )
                     self.socketio.sleep(0)
                     self.current_discussion.update_message(self.current_ai_message_id, self.current_generated_text)
-                    if self.cancel_gen:
-                        ASCIIColors.warning("Generation canceled")
-                        self.cancel_gen = False
-
-
                     # if stop generation is detected then stop
                     if not self.cancel_gen:
                         return True
                     else:
+                        self.cancel_gen = False
+                        ASCIIColors.warning("Generation canceled")
                         return False
             else:
                 self.current_generated_text = self.remove_text_from_string(self.current_generated_text, anti_prompt_to_remove)