diff --git a/tests/simple_tests.py b/tests/simple_tests.py
index e406def93c..0e6ea127d5 100644
--- a/tests/simple_tests.py
+++ b/tests/simple_tests.py
@@ -94,7 +94,7 @@ def train_args(model, model_path, faces, iterations=1, batchsize=2, extra_args="
     """ Train command """
     py_exe = sys.executable
     args = (f"{py_exe} faceswap.py train -A {faces} -B {faces} -m {model_path} -t {model} "
-            f"-b {batchsize} -it {iterations} {extra_args}")
+            f"-b {batchsize} -i {iterations} {extra_args}")
     return args.split()