             button?.setImage(nil, for: .normal)
             button?.setTitle(nil, for: .normal)
             cell.selectionIndex = nil
-            cell.selectionThemeModel = nil
+            updateSelectionThemeModel(nil, for: cell, at: indexPath)
             cell.apply(theme: theme)
         }
         compareToolbarButton.isEnabled = collectionView.indexPathsForSelectedItems?.count ?? 0 == 2

            compareToolbarButton.isEnabled = collectionView.indexPathsForSelectedItems?.count ?? 0 == 2
        }

        guard
            let selectedCell = collectionView.cellForItem(at: indexPath) as? PageHistoryCollectionViewCell
        else {
            return
        }

        let button: UIButton?
        let themeModel: SelectionThemeModel?
        if maxNumberOfRevisionsSelected {
            forEachVisibleCell { (cell: PageHistoryCollectionViewCell?) in
                cell?.selectionThemeModel = self.disabledSelectionThemeModel
                cell?.enableEditing(false)
            }
        }
        switch openSelectionIndex {

        let authorImage: UIImage?
        let sizeDiff: Int?
        let comment: String?
        var selectionThemeModel: SelectionThemeModel?

        init(time: String?, displayTime: String?, author: String?, authorImage: UIImage?, sizeDiff: Int?, comment: String?, selectionThemeModel: SelectionThemeModel?) {
            self.time = time
            self.displayTime = displayTime
            self.author = author
            self.authorImage = authorImage
            self.sizeDiff = sizeDiff
            self.comment = comment
            self.selectionThemeModel = selectionThemeModel
            super.init()
        }
    }
