diff --git a/MVMCoreUI/Atomic/Atoms/Views/Line.swift b/MVMCoreUI/Atomic/Atoms/Views/Line.swift index c22aa11c..62550503 100644 --- a/MVMCoreUI/Atomic/Atoms/Views/Line.swift +++ b/MVMCoreUI/Atomic/Atoms/Views/Line.swift @@ -9,7 +9,9 @@ import UIKit @objcMembers open class Line: View { - var lineModel: LineModel? + var lineModel: LineModel? { + get { return model as? LineModel } + } public var heightConstraint: NSLayoutConstraint? public var widthConstraint: NSLayoutConstraint? @@ -71,7 +73,6 @@ import UIKit open override func set(with model: MoleculeModelProtocol, _ delegateObject: MVMCoreUIDelegateObject?, _ additionalData: [AnyHashable: Any]?) { super.set(with: model, delegateObject, additionalData) if let lineModel = model as? LineModel { - self.lineModel = lineModel setStyle(lineModel.type) } }