diff --git a/Session/Conversations/ConversationVC+Interaction.swift b/Session/Conversations/ConversationVC+Interaction.swift index 33dfce0fb..5d8b3cb95 100644 --- a/Session/Conversations/ConversationVC+Interaction.swift +++ b/Session/Conversations/ConversationVC+Interaction.swift @@ -649,7 +649,7 @@ extension ConversationVC: .elements .firstIndex(of: cellViewModel), let cell = tableView.cellForRow(at: IndexPath(row: index, section: sectionIndex)) as? VisibleMessageCell, - let snapshot = cell.bubbleView.snapshotView(afterScreenUpdates: false), + let snapshot = cell.snContentView.snapshotView(afterScreenUpdates: false), contextMenuWindow == nil, let actions: [ContextMenuVC.Action] = ContextMenuVC.actions( for: cellViewModel, diff --git a/Session/Conversations/Message Cells/VisibleMessageCell.swift b/Session/Conversations/Message Cells/VisibleMessageCell.swift index a7929b69d..d982e1eb6 100644 --- a/Session/Conversations/Message Cells/VisibleMessageCell.swift +++ b/Session/Conversations/Message Cells/VisibleMessageCell.swift @@ -79,7 +79,7 @@ final class VisibleMessageCell: MessageCell, UITextViewDelegate, BodyTextViewDel return result }() - private lazy var snContentView: UIStackView = { + lazy var snContentView: UIStackView = { let result = UIStackView(arrangedSubviews: []) result.axis = .vertical result.spacing = Values.verySmallSpacing