diff --git a/python/examples/imdb/test_client.py b/python/examples/imdb/test_client.py
index 983324a0c47754caa39cb94a61f4ccf1263ee01e..74364e5854d223e380cb386f9a8bc68b8517305a 100644
--- a/python/examples/imdb/test_client.py
+++ b/python/examples/imdb/test_client.py
@@ -23,7 +23,7 @@ client.connect(["127.0.0.1:9292"])
 # you can define any english sentence or dataset here
 # This example reuses imdb reader in training, you
 # can define your own data preprocessing easily.
-imdb_dataset = IMDDataset()
+imdb_dataset = IMDBDataset()
 imdb_dataset.load_resource(sys.argv[2])
 
 for line in sys.stdin: