diff --git a/python/paddle/fluid/async_executor.py b/python/paddle/fluid/async_executor.py
index 2a6a11805e48d209493e1ae579898dc386c3dd92..b077e1be7e4b9bf91f327cd8bb36e5c6ddf85c97 100644
--- a/python/paddle/fluid/async_executor.py
+++ b/python/paddle/fluid/async_executor.py
@@ -153,13 +153,10 @@ class AsyncExecutor(object):
                                      data_feed.desc(), filelist, thread_num,
                                      fetch_var_names, mode, debug)
 
-    def download_data(self, afs_path, local_path, fs_default_name, ugi, process_num=12):
-        #hadoop_home = "$HADOOP_HOME"
+    def download_data(self, afs_path, local_path, fs_default_name, ugi, hadoop_home="$HADOOP_HOME", process_num=12):
         if self.instance is None:
             raise ValueError('instance is None, please run config_distributed_nodes init instance')
             
-        hadoop_home = "~/tools/hadoop-xingtian/hadoop/"
-
         configs = {
             "fs.default.name": fs_default_name,
             "hadoop.job.ugi": ugi
diff --git a/python/setup.py.in b/python/setup.py.in
index 200b96ec54ee5daeb905e155d0b7b57ab7740250..9418804be20e477250d39e64b0ef3ac0955fec4a 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -103,8 +103,10 @@ packages=['paddle',
           'paddle.fluid',
           'paddle.fluid.proto',
           'paddle.fluid.proto.profiler',
+          'paddle.fluid.distributed',
           'paddle.fluid.layers',
           'paddle.fluid.contrib',
+          'paddle.fluid.contrib.utils',
           'paddle.fluid.contrib.decoder',
           'paddle.fluid.contrib.quantize',
           'paddle.fluid.transpiler',