diff --git a/toolkit/config_modules.py b/toolkit/config_modules.py
index 561d97e8..f7111a32 100644
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -18,7 +18,7 @@ if TYPE_CHECKING:
 class SaveConfig:
     def __init__(self, **kwargs):
         self.save_every: int = kwargs.get('save_every', 1000)
-        self.dtype: str = kwargs.get('save_dtype', 'float16')
+        self.dtype: str = kwargs.get('dtype', 'float16')
         self.max_step_saves_to_keep: int = kwargs.get('max_step_saves_to_keep', 5)
         self.save_format: SaveFormat = kwargs.get('save_format', 'safetensors')
         if self.save_format not in ['safetensors', 'diffusers']: