             cell.author = item.user
             cell.sizeDiff = item.revisionSize
             cell.comment = item.parsedComment?.removingHTML
-            if !cell.isSelected {
+            if isSelected, let selectionIndex = indexPathsSelectedForComparisonGroupedByButtonTags.first(where: { $0.value == indexPath })?.key {
+                cell.isSelected = true
+                collectionView.selectItem(at: indexPath, animated: false, scrollPosition: [])
+                cell.selectionThemeModel = selectionIndex == 0 ? firstSelectionThemeModel : secondSelectionThemeModel
+                cell.selectionIndex = selectionIndex
+            } else {
                 cell.selectionThemeModel = maxNumberOfRevisionsSelected ? disabledSelectionThemeModel : nil
             }
         }

                    cell.isSelected = true
                    collectionView.selectItem(at: indexPath, animated: false, scrollPosition: [])
                }
                if cell.isSelected {
                    cell.selectionThemeModel = cachedCellContent.selectionThemeModel
                } else {
                    cell.selectionThemeModel = maxNumberOfRevisionsSelected ? disabledSelectionThemeModel : nil

    var openSelectionIndex = 0

    private var indexPathsSelectedForComparisonGroupedByButtonTags = [Int: IndexPath]() {
        didSet {
            indexPathsSelectedForComparison = Set(indexPathsSelectedForComparisonGroupedByButtonTags.values)
        }
    }
    private var indexPathsSelectedForComparison = Set<IndexPath>()

    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        
