diff --git a/python/paddle_fl/mpc/examples/mnist_demo/mnist_demo.py b/python/paddle_fl/mpc/examples/mnist_demo/mnist_demo.py
index d1afe12d184a5306bd96fa8d3f4910ebe4b85c5e..d441dcb439b859fabaa2f938ba730322256fa890 100644
--- a/python/paddle_fl/mpc/examples/mnist_demo/mnist_demo.py
+++ b/python/paddle_fl/mpc/examples/mnist_demo/mnist_demo.py
@@ -27,11 +27,9 @@ import paddle_fl.mpc as pfl_mpc
 import paddle_fl.mpc.data_utils.aby3 as aby3
 
 
-env_dist = os.environ
-local_host= env_dist.get('LOCALHOST')
 role, server, port = sys.argv[1], sys.argv[2], sys.argv[3]
 # modify host(localhost).
-pfl_mpc.init("aby3", int(role), local_host, server, int(port))
+pfl_mpc.init("aby3", int(role), "localhost", server, int(port))
 role = int(role)
 
 # data preprocessing