2 files changed, 2 insertions, 2 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 48a513c..21fe2fb 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -134,7 +134,7 @@ def parse_args():
     )
     parser.add_argument(
         "--tag_shuffle",
-        type="store_true",
+        action="store_true",
         default=True,
         help="Shuffle tags.",
     )
diff --git a/train_ti.py b/train_ti.py
index 35be74c..b88ccc3 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -170,7 +170,7 @@ def parse_args():
     )
     parser.add_argument(
         "--tag_shuffle",
-        type="store_true",
+        action="store_true",
         help="Shuffle tags.",
     )
     parser.add_argument(