diff --git a/submodules/TelegramUI/TelegramUI/ChatMessageContextControllerContentSource.swift b/submodules/TelegramUI/TelegramUI/ChatMessageContextControllerContentSource.swift
index d59840356c..e7014b23ac 100644
--- a/submodules/TelegramUI/TelegramUI/ChatMessageContextControllerContentSource.swift
+++ b/submodules/TelegramUI/TelegramUI/ChatMessageContextControllerContentSource.swift
@@ -26,7 +26,7 @@ final class ChatMessageContextControllerContentSource: ContextControllerContentS
             guard let item = itemNode.item else {
                 return
             }
-            if item.message.stableId == self.message.stableId, let contentNode = itemNode.getMessageContextSourceNode() {
+            if item.content.contains(where: { $0.stableId == self.message.stableId }), let contentNode = itemNode.getMessageContextSourceNode() {
                 result = ContextControllerTakeViewInfo(contentContainingNode: contentNode, contentAreaInScreenSpace: chatNode.convert(chatNode.frameForVisibleArea(), to: nil))
             }
         }
@@ -46,7 +46,7 @@ final class ChatMessageContextControllerContentSource: ContextControllerContentS
             guard let item = itemNode.item else {
                 return
             }
-            if item.message.stableId == self.message.stableId {
+            if item.content.contains(where: { $0.stableId == self.message.stableId }) {
                 result = ContextControllerPutBackViewInfo(contentAreaInScreenSpace: chatNode.convert(chatNode.frameForVisibleArea(), to: nil))
             }
         }