diff --git a/python/paddle/v2/dataset/imdb.py b/python/paddle/v2/dataset/imdb.py
index 63588c9b31f55e4fc31d896fb6ffe82ea1ea029f..89757e9ef6bbd21bf69236715cffca8efbdb9088 100644
--- a/python/paddle/v2/dataset/imdb.py
+++ b/python/paddle/v2/dataset/imdb.py
@@ -25,6 +25,7 @@ import collections
 import tarfile
 import re
 import string
+import random
 
 __all__ = ['build_dict', 'train', 'test', 'convert']