Merge pull request #2 from Anetsolutions/fix/weak_delegate

Fix memory leak
Fawxy · Jan 30, 2018 · c6486d1 · c6486d1
2 parents c525de1 + 044a130
commit c6486d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CBPinEntryView/Classes/CBPinEntryView.swift b/CBPinEntryView/Classes/CBPinEntryView.swift
@@ -8,7 +8,7 @@
 
 import UIKit
 
-public protocol CBPinEntryViewDelegate {
+public protocol CBPinEntryViewDelegate: class {
     func entryChanged(_ completed: Bool)
 }
 
@@ -81,7 +81,7 @@ public protocol CBPinEntryViewDelegate {
 
     fileprivate var entryButtons: [UIButton] = [UIButton]()
 
-    public var delegate: CBPinEntryViewDelegate?
+    public weak var delegate: CBPinEntryViewDelegate?
 
     override public init(frame: CGRect) {
         super.init(frame: frame)