diff --git a/FSCalendar/FSCalendar.m b/FSCalendar/FSCalendar.m index 6b2c181..0596511 100644 --- a/FSCalendar/FSCalendar.m +++ b/FSCalendar/FSCalendar.m @@ -585,6 +585,9 @@ typedef NS_ENUM(NSUInteger, FSCalendarOrientation) { - (void)collectionView:(UICollectionView *)collectionView willDisplayCell:(UICollectionViewCell *)cell forItemAtIndexPath:(NSIndexPath *)indexPath { + if (![cell isKindOfClass:[FSCalendarCell class]]) { + return; + } NSDate *date = [self.calculator dateForIndexPath:indexPath]; FSCalendarMonthPosition monthPosition = [self.calculator monthPositionForIndexPath:indexPath]; [self.delegateProxy calendar:self willDisplayCell:(FSCalendarCell *)cell forDate:date atMonthPosition:monthPosition];