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 90d73064..1304b90d 100644
--- a/packages/stream_chat_flutter/lib/src/image_actions_modal.dart
+++ b/packages/stream_chat_flutter/lib/src/image_actions_modal.dart
@@ -109,10 +109,7 @@ class ImageActionsModal extends StatelessWidget {
                           final remainingAttachments = [...message.attachments]
                             ..removeAt(currentIndex);
                           channel.updateMessage(message.copyWith(
-                            attachments: remainingAttachments.map((e) {
-                              return e.copyWith(
-                                  uploadState: UploadState.success());
-                            }).toList(),
+                            attachments: remainingAttachments,
                           ));
                           Navigator.pop(context);
                           Navigator.pop(context);