diff --git a/ppcls/utils/check.py b/ppcls/utils/check.py
index bc7030818e67b5997c5f002f27b0e360f8e91867..59255124a6c82a072f351c832ffe16ec6bd269c4 100644
--- a/ppcls/utils/check.py
+++ b/ppcls/utils/check.py
@@ -22,9 +22,7 @@ import sys
 import paddle
 from paddle import is_compiled_with_cuda
 
-from ppcls.arch import get_architectures
-from ppcls.arch import similar_architectures
-from ppcls.arch import get_blacklist_model_in_static_mode
+from ppcls.arch.utils import get_architectures, similar_architectures, get_blacklist_model_in_static_mode
 from ppcls.utils import logger
 
 
diff --git a/ppcls/utils/model_zoo.py b/ppcls/utils/model_zoo.py
index fc527f6a16600f0bfbdd075e3b2de5b466767f28..fce9f4644ca4b918fdcb07b3d8750dcf45b43e6f 100644
--- a/ppcls/utils/model_zoo.py
+++ b/ppcls/utils/model_zoo.py
@@ -23,7 +23,7 @@ import tarfile
 import tqdm
 import zipfile
 
-from ppcls.arch import similar_architectures
+from ppcls.arch.utils import similar_architectures
 from ppcls.utils import logger
 
 __all__ = ['get']