         let authorImage: UIImage?
         let sizeDiff: Int?
         let comment: String?
-
-        init(time: String?, displayTime: String?, author: String?, authorImage: UIImage?, sizeDiff: Int?, 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()
         }
     }

                openSelectionIndex = 0
                navigationItem.rightBarButtonItem = compareButton
                collectionView.indexPathsForSelectedItems?.forEach { collectionView.deselectItem(at: $0, animated: true) }
                forEachVisibleCell { (cell: PageHistoryCollectionViewCell) in
                    cell.selectionThemeModel = nil
                    cell.enableEditing(true) // confusing, have a reset method
                    cell.setEditing(false)
                }

            cell.author = cachedCellContent.author
            cell.sizeDiff = cachedCellContent.sizeDiff
            cell.comment = cachedCellContent.comment
            cell.selectionThemeModel = cachedCellContent.selectionThemeModel
        } else {
            if let date = item.revisionDate {
                if (date as NSDate).wmf_isTodayUTC() {
