diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py
index 0328c172cd02b9016509a1a39b58ee152b3f9554..0faf9d09693433d8f528372cb45ccbb5ab844802 100644
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@@ -878,7 +878,7 @@ class DistributeTranspiler(object):
         origin_param_var = self.origin_program.global_block().vars[
             self.table_name]
 
-        zero_dim = long(
+        zero_dim = int(
             math.ceil(origin_param_var.shape[0] / len(self.pserver_endpoints)))
         table_shape = list(origin_param_var.shape)
         table_shape[0] = zero_dim