diff --git a/paddle/fluid/operators/detail/grpc_server.cc b/paddle/fluid/operators/detail/grpc_server.cc
index 21bd232260f2e080220c41f9a501be71bc5986f2..4079df7ab1099da45136e23a3f6675ae276efa16 100644
--- a/paddle/fluid/operators/detail/grpc_server.cc
+++ b/paddle/fluid/operators/detail/grpc_server.cc
@@ -216,10 +216,10 @@ class RequestCheckpointNotify final : public RequestBase {
   }
 
  protected:
-  sendrecv::CheckpointMessage request_;
+  std::shared_ptr<VariableResponse> request_;
   sendrecv::VoidMessage reply_;
   ServerAsyncResponseWriter<sendrecv::VoidMessage> responder_;
-}
+};
 
 void AsyncGRPCServer::WaitServerReady() {
   VLOG(3) << "AsyncGRPCServer is wait server ready";