diff --git a/src/mflux/config/runtime_config.py b/src/mflux/config/runtime_config.py
index 490f9ba..8b65bca 100644
--- a/src/mflux/config/runtime_config.py
+++ b/src/mflux/config/runtime_config.py
@@ -40,7 +40,7 @@ class RuntimeConfig:
         return self.model_config.num_train_steps
 
     @property
-    def init_image_path(self) -> int:
+    def init_image_path(self) -> str:
         return self.config.init_image_path
 
     @property