diff --git a/packages/stream_chat_flutter/lib/src/image_actions_modal.dart b/packages/stream_chat_flutter/lib/src/image_actions_modal.dart
index 7d004d37..90d73064 100644
--- a/packages/stream_chat_flutter/lib/src/image_actions_modal.dart
+++ b/packages/stream_chat_flutter/lib/src/image_actions_modal.dart
@@ -103,15 +103,20 @@ class ImageActionsModal extends StatelessWidget {
                         color: StreamChatTheme.of(context).colorTheme.accentRed,
                       ),
                       () {
+                        final channel = StreamChannel.of(context).channel;
                         if (message.attachments.length > 1 ||
                             message.text.isNotEmpty) {
-                          var channel = StreamChannel.of(context).channel;
-                          channel.updateMessage(
-                              message..attachments.removeAt(currentIndex));
+                          final remainingAttachments = [...message.attachments]
+                            ..removeAt(currentIndex);
+                          channel.updateMessage(message.copyWith(
+                            attachments: remainingAttachments.map((e) {
+                              return e.copyWith(
+                                  uploadState: UploadState.success());
+                            }).toList(),
+                          ));
                           Navigator.pop(context);
                           Navigator.pop(context);
                         } else {
-                          var channel = StreamChannel.of(context).channel;
                           channel.deleteMessage(message).then((value) {
                             Navigator.pop(context);
                             Navigator.pop(context);