diff --git a/README.md b/README.md
index 860d821..e37f3a1 100644
--- a/README.md
+++ b/README.md
@@ -131,6 +131,8 @@ Then redo the installation instruction within the kohya_ss venv.
 
 ## Change history
 
+* 2023/01/27 (v20.5.1):
+    - Fix issue https://github.com/bmaltais/kohya_ss/issues/71
 * 2023/01/26 (v20.5.0):
     - Add new `Dreambooth TI` tab for training of Textual Inversion embeddings
     - Add Textual Inversion training. Documentation is [here](./train_ti_README-ja.md) (in Japanese.)
diff --git a/lora_gui.py b/lora_gui.py
index 14ab60b..725d4ef 100644
--- a/lora_gui.py
+++ b/lora_gui.py
@@ -159,7 +159,7 @@ def open_configuration(
     stop_text_encoder_training,
     use_8bit_adam,
     xformers,
-    save_model_as_dropdown,
+    save_model_as,
     shuffle_caption,
     save_state,
     resume,