diff --git a/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/Poll/Incoming/PollIncomingBubbleCell.swift b/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/Poll/Incoming/PollIncomingBubbleCell.swift index 61d11e642..c71810ef5 100644 --- a/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/Poll/Incoming/PollIncomingBubbleCell.swift +++ b/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/Poll/Incoming/PollIncomingBubbleCell.swift @@ -16,7 +16,7 @@ import Foundation -class PollIncomingBubbleCell: PollBaseBubbleCell { +class PollIncomingBubbleCell: PollBaseBubbleCell, BubbleIncomingRoomCellProtocol { override func setupViews() { super.setupViews() @@ -27,7 +27,9 @@ class PollIncomingBubbleCell: PollBaseBubbleCell { let messageLeftMargin: CGFloat = 48 + bubbleBackgroundSideMargin bubbleCellContentView?.innerContentViewTrailingConstraint.constant = messageViewMarginRight - bubbleCellContentView?.innerContentViewLeadingConstraint.constant = messageLeftMargin + bubbleCellContentView?.innerContentViewLeadingConstraint.constant = messageLeftMargin + + self.setupBubbleDecorations() } override func update(theme: Theme) { diff --git a/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/VoiceMessage/Outgoing/VoiceMessageOutgoingWithoutSenderInfoBubbleCell.swift b/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/VoiceMessage/Outgoing/VoiceMessageOutgoingWithoutSenderInfoBubbleCell.swift index 1b18d861c..5b0bffd7e 100644 --- a/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/VoiceMessage/Outgoing/VoiceMessageOutgoingWithoutSenderInfoBubbleCell.swift +++ b/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/Cells/VoiceMessage/Outgoing/VoiceMessageOutgoingWithoutSenderInfoBubbleCell.swift @@ -16,7 +16,7 @@ import Foundation -class VoiceMessageOutgoingWithoutSenderInfoBubbleCell: VoiceMessageBubbleCell { +class VoiceMessageOutgoingWithoutSenderInfoBubbleCell: VoiceMessageBubbleCell, BubbleOutgoingRoomCellProtocol { override func setupViews() { super.setupViews() @@ -33,6 +33,8 @@ class VoiceMessageOutgoingWithoutSenderInfoBubbleCell: VoiceMessageBubbleCell { bubbleCellContentView?.innerContentViewLeadingConstraint.constant = leftMargin playbackController.playbackView.stackViewTrailingContraint.constant = playbackViewRightMargin + + self.setupBubbleDecorations() } override func update(theme: Theme) {