diff --git a/VDS/Classes/Control.swift b/VDS/Classes/Control.swift index c0d3dcc8..de2a2dae 100644 --- a/VDS/Classes/Control.swift +++ b/VDS/Classes/Control.swift @@ -94,6 +94,9 @@ open class Control: UIControl, ModelHandlerable, ViewProto open func reset() { backgroundColor = .clear + if let model = model as? Resettable { + model.reset() + } } // MARK: - ViewProtocol diff --git a/VDS/Classes/View.swift b/VDS/Classes/View.swift index 15609f31..20cc0d6b 100644 --- a/VDS/Classes/View.swift +++ b/VDS/Classes/View.swift @@ -94,6 +94,9 @@ open class View: UIView, ModelHandlerable, ViewProtocol, R open func reset() { backgroundColor = .clear + if let model = model as? Resettable { + model.reset() + } } // MARK: - ViewProtocol