         let selectedViewModels = selectedIndexes.compactMap { viewModels.count > $0 ? viewModels[$0] : nil }
         return selectedViewModels
     }
+    
+    func deselectAllCells() {
+        notificationsView.collectionView.indexPathsForSelectedItems?.forEach {
+            notificationsView.collectionView.deselectItem(at: $0, animated: false)
+        }
+    }
 }
 
 // MARK: - NotificationCenterViewModelDelegate

        }
        
        return false
    }
}


    override func setEditing(_ editing: Bool, animated: Bool) {
        super.setEditing(editing, animated: animated)
        
        notificationsView.collectionView.allowsMultipleSelection = editing
        deselectAllCells()
        
        //Reconfigure visible cells to reflect new edit mode.
        //This has a smoother change vs reloadData()
        if #available(iOS 15.0, *) {
            if var snapshot = self.dataSource?.snapshot() {
                snapshot.reconfigureItems(snapshot.itemIdentifiers)
                self.dataSource?.apply(snapshot)
            }
        } else {
            notificationsView.collectionView.visibleCells.forEach { cell in
