diff --git a/paddle/fluid/framework/details/data_balance_op_handle.h b/paddle/fluid/framework/details/data_balance_op_handle.h
index a4adafdfeb19fa9ec57c5b3c5313887e8361d378..76a407e3610e8bb48facf1f814779f4c23f92d98 100644
--- a/paddle/fluid/framework/details/data_balance_op_handle.h
+++ b/paddle/fluid/framework/details/data_balance_op_handle.h
@@ -35,7 +35,7 @@ struct DataBalanceOpHandle : public OpHandleBase {
                       const platform::NCCLContextMap *ctxs);
 #else
   DataBalanceOpHandle(const std::vector<Scope *> &local_scopes,
-                      const std::vector<platform::Place> *places)
+                      const std::vector<platform::Place> &places);
 #endif
 
   std::string Name() const override;