diff --git a/tests/test_runner/test_runner.py b/tests/test_runner/test_runner.py
index 6b984c57..5005927a 100644
--- a/tests/test_runner/test_runner.py
+++ b/tests/test_runner/test_runner.py
@@ -227,7 +227,8 @@ class CustomRunner(Runner):
                  train_cfg=None,
                  val_cfg=None,
                  test_cfg=None,
-                 optimizer=None,
+                 auto_scale_lr_cfg=None,
+                 optim_wrapper=None,
                  param_scheduler=None,
                  val_evaluator=None,
                  test_evaluator=None,