diff --git a/submodules/TelegramUI/TelegramUI/ChatMessageBubbleItemNode.swift b/submodules/TelegramUI/TelegramUI/ChatMessageBubbleItemNode.swift
index f6be80912a..88a19d0bdb 100644
--- a/submodules/TelegramUI/TelegramUI/ChatMessageBubbleItemNode.swift
+++ b/submodules/TelegramUI/TelegramUI/ChatMessageBubbleItemNode.swift
@@ -76,7 +76,12 @@ private func contentNodeMessagesAndClassesForItem(_ item: ChatMessageItem) -> [(
             }
         }
         
-        if !message.text.isEmpty || isUnsupportedMedia {
+        var messageText = message.text
+        if let updatingMedia = itemAttributes.updatingMedia {
+            messageText = updatingMedia.text
+        }
+        
+        if !messageText.isEmpty || isUnsupportedMedia {
             if !skipText {
                 if case .group = item.content {
                     messageWithCaptionToAdd = (message, itemAttributes)