diff --git a/MVMCoreUI/BaseClasses/ImageView.swift b/MVMCoreUI/BaseClasses/ImageView.swift
index b2707ec2..aa004d27 100644
--- a/MVMCoreUI/BaseClasses/ImageView.swift
+++ b/MVMCoreUI/BaseClasses/ImageView.swift
@@ -54,8 +54,8 @@ open class ImageView: UIImageView, ModelMoleculeViewProtocol {
     //--------------------------------------------------
     // MARK: - ModelMoleculeViewProtocol
     //--------------------------------------------------
-
-    open func set(with model: MoleculeModelProtocol?, _ delegateObject: MVMCoreUIDelegateObject?, _ additionalData: [AnyHashable: Any]?) {
+    
+    public func set(with model: MoleculeModelProtocol, _ delegateObject: MVMCoreUIDelegateObject?, _ additionalData: [AnyHashable : Any]?) {
         self.model = model
         if let backgroundColor = model?.backgroundColor {
             self.backgroundColor = backgroundColor.uiColor