Send: reset composer directly

features/messages/src/main/java/io/element/android/x/features/messages/MessagesViewModel.kt

@@ -68,8 +68,10 @@ class MessagesViewModel(
     }
 
     fun sendMessage(text: String) {
-        withState { state ->
         viewModelScope.launch {
+            val state = awaitState()
+            // Reset composer right away
+            setNormalMode()
             when (state.composerMode) {
                 is MessageComposerMode.Normal -> timeline.sendMessage(text)
                 is MessageComposerMode.Edit -> timeline.editMessage(
@@ -82,9 +84,6 @@ class MessagesViewModel(
                     text
                 )
             }
-                // Reset composer
-                setNormalMode()
-            }
         }
     }