diff --git a/src/hssm/hssm.py b/src/hssm/hssm.py
index db29ff49..ad50d432 100644
--- a/src/hssm/hssm.py
+++ b/src/hssm/hssm.py
@@ -1495,7 +1495,7 @@ def save_model(
         base_path : str | Path
             Base directory to save model files in.
             Must be relative path if allow_absolute_base_path=False
-        save_class : bool
+        save_idata_only: bool = False,
             Whether to save the model class instance itself
 
         Raises