diff --git a/python/examples/imdb/text_classify_service.py b/python/examples/imdb/text_classify_service.py
index bbf63bb0cf072f375fb54b8d5dd2ff6386518ae8..50d0d1aebba34a630c16442c6e3d00460bb1bc6a 100755
--- a/python/examples/imdb/text_classify_service.py
+++ b/python/examples/imdb/text_classify_service.py
@@ -29,7 +29,7 @@ class IMDBService(WebService):
         if "words" not in feed:
             exit(-1)
         res_feed = {}
-        res_feed["words"] = self.dataset.get_words_only(feed["words"])[0]
+        res_feed["words"] = self.dataset.get_words_only(feed["words"])
         return res_feed, fetch