diff --git a/text_classification/train.py b/text_classification/train.py
index cda04bfc6a33ee9e39298910d724bc716f1b53df..888fde356f3aec1addb5e5fcf35e17d0c82f37c3 100644
--- a/text_classification/train.py
+++ b/text_classification/train.py
@@ -46,10 +46,10 @@ def train(topology,
         word_dict = paddle.dataset.imdb.word_dict()
         train_reader = paddle.batch(
             paddle.reader.shuffle(
-                lambda: paddle.dataset.imdb.train(word_dict), buf_size=1000),
+                lambda: paddle.dataset.imdb.train(word_dict)(), buf_size=1000),
             batch_size=100)
         test_reader = paddle.batch(
-            lambda: paddle.dataset.imdb.test(word_dict), batch_size=100)
+            lambda: paddle.dataset.imdb.test(word_dict)(), batch_size=100)
 
         class_num = 2
     else: