Попытка заставить демонстрационный код Chatto работать в моем собственном приложении и получить ошибку подтверждения

Question

Попытка заставить демонстрационный код Chatto работать в моем собственном приложении и получить ошибку подтверждения.

https://github.com/badoo/Chatto/tree/master/ChattoApp/ChattoApp/Source

Для минимума я пытаюсь вернуть TextMessagePresenterBuilder

  var presenterBuilders: [ChatItemType: [ChatItemPresenterBuilderProtocol]] {

    let textMessagePresenter = TextMessagePresenterBuilder(
      viewModelBuilder: textMessageViewModelBuilder,
      interactionHandler: GenericMessageHandler(baseHandler: baseMessageHandler)
    )

    textMessagePresenter.baseMessageStyle = BaseMessageCollectionViewCellAvatarStyle()

    return [
      DemoTextMessageModel.chatItemType: [textMessagePresenter]
    ]

  }

Я получаю ошибку подтверждения в heightForCell в BaseMessagePresenter

guard let decorationAttributes = decorationAttributes as? ChatItemDecorationAttributes else {
    assert(false, "Expecting decoration attributes")
    return 0
}

Насколько я понимаю, вам не нужно использовать декоратор, и этот код прямо из демо. Чего мне не хватает?