diff --git a/README.md b/README.md
index a8230d3..7f1e268 100644
--- a/README.md
+++ b/README.md
@@ -35,11 +35,11 @@ Please use BertTokenizerFast as tokenizer instead of AutoTokenizer.
 ```
 from transformers import (
   BertTokenizerFast,
-  AutoModelForTokenClassification,
+  AutoModel,
 )
 
 tokenizer = BertTokenizerFast.from_pretrained('bert-base-chinese')
-model = AutoModelForTokenClassification.from_pretrained('ckiplab/bert-base-chinese-ner')
+model = AutoModel.from_pretrained('ckiplab/bert-base-chinese-ner')
 ```
 
 For full usage and more information, please refer to https://github.com/ckiplab/ckip-transformers.