 
             dataSorted[idx] = AnyChatSection(MessageSection(
                 object: AnyDifferentiable(section),
-                controllerContext: controllerContext
+                controllerContext: controllerContext,
+                collapsibleItemsState: collpsibleItemsState
             ))
 
             if unreadMarker {

            guard let messageSection1 = object.object.base as? MessageSectionModel else { continue }

            let message = messageSection1.message
            let unreadMarker = !hasUnreadMarker && message.createdAt > lastSeen
            var separator: Date?
            var sequential = false


        return AnyChatSection(MessageSection(
            object: AnyDifferentiable(messageSectionModel),
            controllerContext: controllerContext,
            collapsibleItemsState: [:]
        ))
    }

