diff --git a/PaddleNLP/sentiment_classification/run_ernie_classifier.py b/PaddleNLP/sentiment_classification/run_ernie_classifier.py
index 21ab742616d4cf62f63583df7ede742be87138ef..f667aa68398ad7e66b79244aeb4a16915da35852 100644
--- a/PaddleNLP/sentiment_classification/run_ernie_classifier.py
+++ b/PaddleNLP/sentiment_classification/run_ernie_classifier.py
@@ -38,16 +38,16 @@ from utils import init_checkpoint
 
 
 def ernie_pyreader(args, pyreader_name):
-    src_ids = fluid.data(
+    src_ids = fluid.layers.data(
         name="src_ids", shape=[None, args.max_seq_len, 1], dtype="int64")
-    sent_ids = fluid.data(
+    sent_ids = fluid.layers.data(
         name="sent_ids", shape=[None, args.max_seq_len, 1], dtype="int64")
-    pos_ids = fluid.data(
+    pos_ids = fluid.layers.data(
         name="pos_ids", shape=[None, args.max_seq_len, 1], dtype="int64")
-    input_mask = fluid.data(
+    input_mask = fluid.layers.data(
         name="input_mask", shape=[None, args.max_seq_len, 1], dtype="float32")
-    labels = fluid.data(name="labels", shape=[None, 1], dtype="int64")
-    seq_lens = fluid.data(name="seq_lens", shape=[None], dtype="int64")
+    labels = fluid.layers.data(name="labels", shape=[None, 1], dtype="int64")
+    seq_lens = fluid.layers.data(name="seq_lens", shape=[None], dtype="int64")
 
     pyreader = fluid.io.DataLoader.from_generator(
         feed_list=[src_ids, sent_ids, pos_ids, input_mask, labels, seq_lens],