diff --git a/python/paddle/fluid/tests/unittests/test_communicator_async3.py b/python/paddle/fluid/tests/unittests/test_communicator_async3.py
index f53dcb920e629cdbc722000e7cc68d94bfb96ac2..2e4df9a0757cd28e062777b3aeccc60fca23db25 100644
--- a/python/paddle/fluid/tests/unittests/test_communicator_async3.py
+++ b/python/paddle/fluid/tests/unittests/test_communicator_async3.py
@@ -40,7 +40,7 @@ class TestCommunicator(unittest.TestCase):
     def test_communicator_async(self):
         role = role_maker.UserDefinedRoleMaker(
             current_id=0,
-            role=role_maker.Role.WORKER,
+            role=role_maker.Role.SERVER,
             worker_num=2,
             server_endpoints=["127.0.0.1:6001", "127.0.0.1:6002"])
 
@@ -52,10 +52,6 @@ class TestCommunicator(unittest.TestCase):
         optimizer = fleet.distributed_optimizer(optimizer, strategy)
         optimizer.minimize(avg_cost)
 
-        fleet.init_worker()
-        time.sleep(10)
-        fleet.stop_worker()
-
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py b/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py
index dc40b2eb5c6480fb22f28c66e2b8205575269b66..d398bd073dc0cea119f430a14ff596e4cd586544 100644
--- a/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py
@@ -164,7 +164,7 @@ class TestPSPassWithBow(unittest.TestCase):
 
         fleet.init(role)
         loss, acc, _ = self.net()
-        optimizer = fluid.optimizer.SGD(base_lr)
+        optimizer = fluid.optimizer.Adam(base_lr)
         strategy = StrategyFactory.create_async_strategy()
         optimizer = fleet.distributed_optimizer(optimizer, strategy)
         optimizer.minimize(loss)