diff --git a/Sources/ContainerController/ContainerController.swift b/Sources/ContainerController/ContainerController.swift index 3b5a13f..8863ab3 100644 --- a/Sources/ContainerController/ContainerController.swift +++ b/Sources/ContainerController/ContainerController.swift @@ -17,7 +17,7 @@ open class ContainerController: NSObject { public var shadowButton: UIButton! - public var controller: UIViewController? + public weak var controller: UIViewController? public var scrollView: UIScrollView? @@ -31,7 +31,7 @@ open class ContainerController: NSObject { // MARK: Delegate - public var delegate: ContainerControllerDelegate? + public weak var delegate: ContainerControllerDelegate? // MARK: Current Move Type