diff --git a/.tools/build_docker.sh b/.tools/build_docker.sh
index 269687c3e3be4afad1e492cdba9214b6bac1a435..fff09617411904158a0ae7ea4bad9cbbafb842e2 100755
--- a/.tools/build_docker.sh
+++ b/.tools/build_docker.sh
@@ -25,10 +25,7 @@ MAINTAINER PaddlePaddle Authors <paddle-dev@baidu.com>
 
 COPY . /book
 
-#RUN pip install -U nltk \
-#    && python /book/.tools/cache_dataset.py
-
-RUN python -c "import paddle.v2.dataset as dataset; dataset.fetch_all()"
+RUN python -c "import paddle.v2.dataset.common as common; common.fetch_all()"
 
 RUN ${update_mirror_cmd}
     apt-get update && \