diff --git a/MVMCoreUI/Molecules/LeftRightViews/CornerLabels.swift b/MVMCoreUI/Molecules/LeftRightViews/CornerLabels.swift
index bde3ab6f..5c0dbfcf 100644
--- a/MVMCoreUI/Molecules/LeftRightViews/CornerLabels.swift
+++ b/MVMCoreUI/Molecules/LeftRightViews/CornerLabels.swift
@@ -8,7 +8,7 @@
 
 import UIKit
 
-@objcMembers public class CornerLabels: ViewConstrainingView {
+@objcMembers public class CornerLabels: ViewConstrainingView, ModelMoleculeViewProtocol {
     
     let topLeftLabel = Label.commonLabelB1(true)
     let topRightLabel = Label.commonLabelB1(true)
@@ -177,9 +177,7 @@ import UIKit
     public override class func estimatedHeight(forRow json: [AnyHashable : Any]?, delegateObject: MVMCoreUIDelegateObject?) -> CGFloat {
         return 34
     }
-}
 
-extension CornerLabels: MoleculeViewProtocol {
     public func setWithModel(_ model: MoleculeProtocol?, _ delegateObject: MVMCoreUIDelegateObject?, _ additionalData: [AnyHashable: Any]?) {
         guard let model = model as? CornerLabelsModel,
         let data = try? model.encode(using: JSONEncoder()),