diff --git a/model_zoo/lstm/eval.py b/model_zoo/lstm/eval.py
index a9b81199c1e..6d731fbd0df 100644
--- a/model_zoo/lstm/eval.py
+++ b/model_zoo/lstm/eval.py
@@ -23,8 +23,8 @@ import numpy as np
 
 from src.config import lstm_cfg as cfg
 from src.dataset import lstm_create_dataset, convert_to_mindrecord
+from src.lstm import SentimentNet
 from mindspore import Tensor, nn, Model, context
-from mindspore.model_zoo.lstm import SentimentNet
 from mindspore.nn import Accuracy
 from mindspore.train.callback import LossMonitor
 from mindspore.train.serialization import load_checkpoint, load_param_into_net
diff --git a/model_zoo/lstm/train.py b/model_zoo/lstm/train.py
index 732655f1deb..51ae12c6854 100644
--- a/model_zoo/lstm/train.py
+++ b/model_zoo/lstm/train.py
@@ -24,8 +24,8 @@ import numpy as np
 from src.config import lstm_cfg as cfg
 from src.dataset import convert_to_mindrecord
 from src.dataset import lstm_create_dataset
+from src.lstm import SentimentNet
 from mindspore import Tensor, nn, Model, context
-from mindspore.model_zoo.lstm import SentimentNet
 from mindspore.nn import Accuracy
 from mindspore.train.callback import LossMonitor, CheckpointConfig, ModelCheckpoint, TimeMonitor
 from mindspore.train.serialization import load_param_into_net, load_checkpoint