diff --git a/python/paddle/fluid/tests/unittests/test_fleet_base.py b/python/paddle/fluid/tests/unittests/test_fleet_base.py
index 8019841f72eccd35cb76ece582f89179a880e892..2292b06303e8f53c1a4fe994e792872f5fcd25a5 100644
--- a/python/paddle/fluid/tests/unittests/test_fleet_base.py
+++ b/python/paddle/fluid/tests/unittests/test_fleet_base.py
@@ -145,9 +145,8 @@ class TestFleetBase(unittest.TestCase):
         import paddle.fluid.incubate.fleet.base.role_maker as role_maker
         role = role_maker.PaddleCloudRoleMaker(is_collective=True)
         fleet.init(role)
-        strategy = fleet.DistributedStrategy()
         optimizer = paddle.optimizer.SGD(learning_rate=0.001)
-        optimizer = fleet.distributed_optimizer(optimizer, strategy=strategy)
+        optimizer = fleet.distributed_optimizer(optimizer)
 
     def test_minimize(self):
         import paddle