diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py
index 0b79f6be9..d19e82e9f 100644
--- a/dpgen/generator/run.py
+++ b/dpgen/generator/run.py
@@ -400,11 +400,11 @@ def make_train(iter_index, jdata, mdata):
                     nframes += dpdata.LabeledSystem(
                         sys_single, fmt="deepmd/npy"
                     ).get_nframes()
-                    if auto_ratio:
-                        if ii == iter_index - 1:
-                            number_new_frames += nframes
-                        else:
-                            number_old_frames += nframes
+                if auto_ratio:
+                    if ii == iter_index - 1:
+                        number_new_frames += nframes
+                    else:
+                        number_old_frames += nframes
                 if nframes < fp_task_min:
                     log_task(
                         "nframes (%d) in data sys %s is too small, skip" % (nframes, jj)