diff --git a/test/models/test_bert.py b/test/models/test_bert.py
index 38a16f9b..05ee6d5a 100644
--- a/test/models/test_bert.py
+++ b/test/models/test_bert.py
@@ -8,7 +8,7 @@ from fastNLP.models.bert import *
 class TestBert(unittest.TestCase):
     def test_bert_1(self):
         from fastNLP.core.const import Const
-        from fastNLP.modules.encoder._bert import BertConfig
+        from fastNLP.modules.encoder.bert import BertConfig
 
         model = BertForSequenceClassification(2, BertConfig(32000))
 
@@ -23,7 +23,7 @@ class TestBert(unittest.TestCase):
 
     def test_bert_2(self):
         from fastNLP.core.const import Const
-        from fastNLP.modules.encoder._bert import BertConfig
+        from fastNLP.modules.encoder.bert import BertConfig
 
         model = BertForMultipleChoice(2, BertConfig(32000))
 
@@ -38,7 +38,7 @@ class TestBert(unittest.TestCase):
 
     def test_bert_3(self):
         from fastNLP.core.const import Const
-        from fastNLP.modules.encoder._bert import BertConfig
+        from fastNLP.modules.encoder.bert import BertConfig
 
         model = BertForTokenClassification(7, BertConfig(32000))
 
@@ -53,7 +53,7 @@ class TestBert(unittest.TestCase):
 
     def test_bert_4(self):
         from fastNLP.core.const import Const
-        from fastNLP.modules.encoder._bert import BertConfig
+        from fastNLP.modules.encoder.bert import BertConfig
 
         model = BertForQuestionAnswering(BertConfig(32000))