diff --git a/cnocr/cn_ocr.py b/cnocr/cn_ocr.py
index 794fe74d5e37b4b18ae476da9b05a9b3be0b8db2..fef8e6dcc2f8bff9410d29cee38c3a0c286ed652 100644
--- a/cnocr/cn_ocr.py
+++ b/cnocr/cn_ocr.py
@@ -21,7 +21,7 @@ import mxnet as mx
 import numpy as np
 from PIL import Image
 
-from cnocr import MODEL_VERSION, AVAILABLE_MODELS
+from cnocr.consts import MODEL_VERSION, AVAILABLE_MODELS
 from cnocr.hyperparams.cn_hyperparams import CnHyperparams as Hyperparams
 from cnocr.fit.lstm import init_states
 from cnocr.fit.ctc_metrics import CtcMetrics
diff --git a/tests/test_cnocr.py b/tests/test_cnocr.py
index 3eb15264049e2a7f4b984c62830782551cb8a34d..f65eead0723fd045b2236c2ad4ee2fe05b282b4b 100644
--- a/tests/test_cnocr.py
+++ b/tests/test_cnocr.py
@@ -177,7 +177,7 @@ def test_gray_aug(img_fp, expected):
 
 
 def test_cand_alphabet():
-    from cnocr.consts import NUMBERS
+    from cnocr import NUMBERS
 
     img_fp = os.path.join(example_dir, 'hybrid.png')