diff --git a/src/mflux/config/runtime_config.py b/src/mflux/config/runtime_config.py
index 59d2d58..490f9ba 100644
--- a/src/mflux/config/runtime_config.py
+++ b/src/mflux/config/runtime_config.py
@@ -44,7 +44,7 @@ class RuntimeConfig:
         return self.config.init_image_path
 
     @property
-    def init_image_strength(self) -> int:
+    def init_image_strength(self) -> float:
         return self.config.init_image_strength
 
     @property