Fixed memory leak

#333
patchthecode · Mar 7, 2017 · dfeae9a · dfeae9a
1 parent 3bc873c
commit dfeae9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/JTAppleCalendarLayout.swift b/Sources/JTAppleCalendarLayout.swift
@@ -47,7 +47,7 @@ open class JTAppleCalendarLayout: UICollectionViewLayout, JTAppleCalendarLayoutP
         }
     }
 
-    unowned var delegate: JTAppleCalendarDelegateProtocol
+    weak var delegate: JTAppleCalendarDelegateProtocol!
     var currentHeader: (section: Int, size: CGSize)? // Tracks the current header size
     var currentCell: (section: Int, itemSize: CGSize)? // Tracks the current cell size
     var contentHeight: CGFloat = 0 // Content height of calendarView
@@ -57,8 +57,8 @@ open class JTAppleCalendarLayout: UICollectionViewLayout, JTAppleCalendarLayoutP
     var daysInSection: [Int: Int] = [:] // temporary caching
 
     init(withDelegate delegates: JTAppleCalendarDelegateProtocol) {
-        delegate = delegates
         super.init()
+        delegate = delegates
 
     }