diff --git a/MVMCoreUI/Atomic/Atoms/Buttons/ImageButton.swift b/MVMCoreUI/Atomic/Atoms/Buttons/ImageButton.swift
index 323236a7..7dd08e88 100644
--- a/MVMCoreUI/Atomic/Atoms/Buttons/ImageButton.swift
+++ b/MVMCoreUI/Atomic/Atoms/Buttons/ImageButton.swift
@@ -33,7 +33,7 @@ import Foundation
         
         super.set(with: model, delegateObject, additionalData)
         FormValidator.setupValidation(for: castModel, delegate: delegateObject?.formHolderDelegate)
-
+        setState()
     }
     
     public func setState() {