Fix willDisplay

Mu-L · Mar 1, 2017 · 7ac5cb9 · 7ac5cb9
1 parent 3a76fa9
commit 7ac5cb9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/FSCalendar/FSCalendar.m b/FSCalendar/FSCalendar.m
@@ -585,6 +585,9 @@ - (void)collectionView:(UICollectionView *)collectionView didDeselectItemAtIndex
 
 - (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];