diff --git a/colossalai/cli/benchmark/__init__.py b/colossalai/cli/benchmark/__init__.py
index c020d33b6..618ff8c61 100644
--- a/colossalai/cli/benchmark/__init__.py
+++ b/colossalai/cli/benchmark/__init__.py
@@ -1,9 +1,10 @@
 import click
 
-from .utils import *
-from .benchmark import run_benchmark
 from colossalai.context import Config
 
+from .benchmark import run_benchmark
+from .utils import *
+
 __all__ = ['benchmark']