diff --git a/python/examples/imdb/test_ensemble_client.py b/python/examples/imdb/test_ensemble_client.py
index e3723caad6430424c00628949e3c6368a4fa34c9..6cafb3389fff5a25103bcb2b3a867b73b35b9e8e 100644
--- a/python/examples/imdb/test_ensemble_client.py
+++ b/python/examples/imdb/test_ensemble_client.py
@@ -14,7 +14,7 @@
 # pylint: disable=doc-string-missing
 
 from paddle_serving_client import Client
-from paddle_serving_app import IMDBDataset
+from imdb_reader import IMDBDataset
 
 client = Client()
 # If you have more than one model, make sure that the input