diff --git a/Source/CollectionView/Internal/CollectionViewCellWrapper.swift b/Source/CollectionView/Internal/CollectionViewCellWrapper.swift index b5add1b..3b20341 100644 --- a/Source/CollectionView/Internal/CollectionViewCellWrapper.swift +++ b/Source/CollectionView/Internal/CollectionViewCellWrapper.swift @@ -70,6 +70,7 @@ public final class CollectionViewCellWrapper: UICollectionViewCell (cell as? Configurable)?.configuration = configuration self.cell = cell contentView.children(cell) + syncCollectionViewCellState() setNeedsUpdateConstraints() return cell } @@ -81,4 +82,10 @@ public final class CollectionViewCellWrapper: UICollectionViewCell self.cell = nil return cell } + + private func syncCollectionViewCellState() { + // Synchronize the state of the wrapper with the wrapped cell + collectionViewCell?.setSelected(isSelected) + collectionViewCell?.setHighlighted(isHighlighted) + } }