diff --git a/src/apps/chat/components/message/ChatMessage.tsx b/src/apps/chat/components/message/ChatMessage.tsx
index 754f9daa3..a83fe5eff 100644
--- a/src/apps/chat/components/message/ChatMessage.tsx
+++ b/src/apps/chat/components/message/ChatMessage.tsx
@@ -221,10 +221,10 @@ export function ChatMessage(props: {
         handleFragmentReplace(fragmentId, createTextContentFragment(editedText));
       else
         handleFragmentDelete(fragmentId);
-      // if the user pressd Ctrl, we begin a regeneration from here
-      if (withControl && onMessageAssistantFrom)
-        await onMessageAssistantFrom(messageId, 0);
     }
+    // if the user pressd Ctrl, we begin a regeneration from here
+    if (withControl && onMessageAssistantFrom)
+      await onMessageAssistantFrom(messageId, 0);
   }, [handleFragmentDelete, handleFragmentReplace, messageId, onMessageAssistantFrom, textContentEditState]);
 
   const handleEditsApplyClicked = React.useCallback(() => handleApplyAllEdits(false), [handleApplyAllEdits]);