diff --git a/python/paddle/fluid/distributed/helper.py b/python/paddle/fluid/distributed/helper.py
index 999c8d77b83b6fd6629d9b91a75c339ef7f3cad5..cdde5403cda3cc82296c70226cd9962de1482308 100644
--- a/python/paddle/fluid/distributed/helper.py
+++ b/python/paddle/fluid/distributed/helper.py
@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import ps_pb2 as pslib
-
 
 class FileSystem(object):
     """
@@ -37,6 +35,7 @@ class FileSystem(object):
         assert user != None
         assert passwd != None
         assert hadoop_bin != None
+        import ps_pb2 as pslib
         self.fs_client = pslib.FsClientParameter()
         #if fs_type == "afs":
         #    fs_client.fs_type = pslib.FsApiType.AFS