diff --git a/lib/src/message_input.dart b/lib/src/message_input.dart
index 6358e39e..e0e3104b 100644
--- a/lib/src/message_input.dart
+++ b/lib/src/message_input.dart
@@ -909,9 +909,9 @@ class MessageInputState extends State<MessageInput> {
           );
     }
 
-    return sendingFuture.whenComplete(() {
+    return sendingFuture.then((resp) {
       if (widget.onMessageSent != null) {
-        widget.onMessageSent(message);
+        widget.onMessageSent(resp.message);
       } else {
         if (widget.editMessage != null) {
           Navigator.pop(context);