diff --git a/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/BubbleRoomCellLayoutUpdater.swift b/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/BubbleRoomCellLayoutUpdater.swift
index 874be05e3..af7360cd2 100644
--- a/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/BubbleRoomCellLayoutUpdater.swift
+++ b/Riot/Modules/Room/Views/BubbleCells/Styles/Bubble/BubbleRoomCellLayoutUpdater.swift
@@ -196,7 +196,7 @@ class BubbleRoomCellLayoutUpdater: RoomCellLayoutUpdaterProtocol {
         let firstEventId = firstComponent.event.eventId
         let firstMessageTopPosition = cell.topPosition(ofEvent: firstEventId)
             
-        let additionalContentHeight = roomBubbleCellData.additionalHeight(forEvent: lastEventId)
+        let additionalContentHeight = roomBubbleCellData.additionalContentHeight
                 
         bubbleHeight = lastMessageBottomPosition - firstMessageTopPosition - additionalContentHeight + bottomMargin