diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py
index 0fac599847..5a0fd435f7 100644
--- a/deepmd/pt/entrypoints/main.py
+++ b/deepmd/pt/entrypoints/main.py
@@ -474,8 +474,8 @@ def change_bias(
             data_single.systems,
             data_single.dataloaders,
             nbatches,
-            min_frames_per_element_forstat = min_frames,
-            enable_element_completion = not skip_elem_check,
+            min_frames_per_element_forstat=min_frames,
+            enable_element_completion=not skip_elem_check,
         )
         updated_model = training.model_change_out_bias(
             model_to_change, sampled_data, _bias_adjust_mode=bias_adjust_mode
diff --git a/deepmd/utils/argcheck.py b/deepmd/utils/argcheck.py
index 79e763cf4e..122759541c 100644
--- a/deepmd/utils/argcheck.py
+++ b/deepmd/utils/argcheck.py
@@ -2787,9 +2787,7 @@ def training_args(
         "If the gradient norm exceeds this value, it will be clipped to this limit. "
         "No gradient clipping will occur if set to 0."
     )
-    doc_min_frames_per_element_forstat = (
-        "The minimum number of frames per element used for statistics when using the mixed type." 
-    )
+    doc_min_frames_per_element_forstat = "The minimum number of frames per element used for statistics when using the mixed type."
     doc_enable_element_completion = (
         "Whether to check elements when using the mixed type"
     )