diff --git a/reproduction/sequence_labelling/ner/model/dilated_cnn.py b/reproduction/sequence_labelling/ner/model/dilated_cnn.py
index 89d51d56..a691560a 100644
--- a/reproduction/sequence_labelling/ner/model/dilated_cnn.py
+++ b/reproduction/sequence_labelling/ner/model/dilated_cnn.py
@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from fastNLP.modules.decoder import ConditionalRandomField
-from fastNLP.modules.encoder import Embedding
+from fastNLP.embeddings import Embedding
 from fastNLP.core.utils import seq_len_to_mask
 from fastNLP.core.const import Const as C
 
diff --git a/reproduction/sequence_labelling/ner/model/lstm_cnn_crf.py b/reproduction/sequence_labelling/ner/model/lstm_cnn_crf.py
index c38dce38..1d51ab79 100644
--- a/reproduction/sequence_labelling/ner/model/lstm_cnn_crf.py
+++ b/reproduction/sequence_labelling/ner/model/lstm_cnn_crf.py
@@ -1,5 +1,4 @@
 
-import torch
 from torch import nn
 from fastNLP import seq_len_to_mask
 from fastNLP.modules import LSTM