Место для заголовка раздела - MessageKit Chat Swift

Question

Я устанавливаю cellTopLabelAttributedText с датой, только если дни не совпадают, иначе я не хочу лишний пробел между ячейками. Как установить cellTopLabelHeight только для некоторых ячеек?

как это выглядит

как и должно быть

Какой-то код:

func cellTopLabelAttributedText(for message: MessageType, at indexPath: IndexPath) -> NSAttributedString? {
    if !Calendar.current.isDate(lastDate!, inSameDayAs: message.sentDate) {
        lastDate = message.sentDate
        return NSAttributedString(string: date, attributes: [NSAttributedString.Key.font: font!, NSAttributedString.Key.foregroundColor: color!])
    }
    return nil
}

func cellTopLabelHeight(for message: MessageType, at indexPath: IndexPath, in messagesCollectionView: MessagesCollectionView) -> CGFloat {
    return 50
}

Я пытался использовать один и тот же оператор if в обеих функциях, но он не работал.