undod reload change

patchthecode · Jul 17, 2017 · 215fa91 · 215fa91
1 parent 9d475f0
commit 215fa91
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Sources/JTAppleCalendarView.swift b/Sources/JTAppleCalendarView.swift
@@ -116,7 +116,8 @@ open class JTAppleCalendarView: UICollectionView {
         if (collectionViewLayout as! JTAppleCalendarLayout).lastSetCollectionViewSize != frame {
             // ive seen that layout subview gets called. Calendar is setup, and then it gets called again
             // At this point, it already has generaed cells which needs to be invalidated.
-            reloadData()
+            calendarViewLayout.invalidateLayout()
+            layoutIfNeeded()
         }
 
         if !delayedExecutionClosure.isEmpty, isCalendarLayoutLoaded {