diff --git a/MVMCoreUI/Atoms/Views/MFLoadImageView.swift b/MVMCoreUI/Atoms/Views/MFLoadImageView.swift
index fa7d369e..156e187d 100644
--- a/MVMCoreUI/Atoms/Views/MFLoadImageView.swift
+++ b/MVMCoreUI/Atoms/Views/MFLoadImageView.swift
@@ -210,6 +210,9 @@ import UIKit
     }
     
     public func setWithModel(_ model: MoleculeModelProtocol?, _ delegateObject: MVMCoreUIDelegateObject?, _ additionalData: [AnyHashable: Any]?) {
+        self.delegateObject = delegateObject
+        // TODO: Temporary, should be moved to init once we have type erasure ready.
+        setAsMolecule()
         guard let imageModel = model as? ImageViewModel else {
             return
         }