diff --git a/paddle/fluid/operators/detail/grpc_client.cc b/paddle/fluid/operators/detail/grpc_client.cc
index 17476ab513b55226f0b6db622841fda80850bda8..7a63e39d5aae6fc98f220d7c3f3ab181d7397ee9 100644
--- a/paddle/fluid/operators/detail/grpc_client.cc
+++ b/paddle/fluid/operators/detail/grpc_client.cc
@@ -236,17 +236,10 @@ void GRPCClient::AsyncCheckpointNotify(const std::string& ep,
 
   CheckpointNotifyProcessor* s = new CheckpointNotifyProcessor(ch);
   s->Prepare(time_out);
-  s->response_call_back_ = nullptr;
 
   sendrecv::VariableMessage req;
   req.set_varname(CHECKPOINT_SAVE_MESSAGE);
-  req.out_varname(dir);
-
-  auto call = s->stub_g_.PrepareUnaryCall(
-      s->context_.get(), "/sendrecv.SendRecvService/CheckpointNotify", req,
-      &cq_);
-  call->StartCall();
-  call->Finish(&s->reply_, &s->status_, reinterpret_cast<void*>(s));
+  req.set_out_varname(dir);
 
   auto rpc = s->stub_->AsyncCheckpointNotify(s->context_.get(), req, &cq_);
   rpc->Finish(&s->reply_, &s->status_, reinterpret_cast<void*>(s));
diff --git a/paddle/fluid/operators/listen_and_serv_op.cc b/paddle/fluid/operators/listen_and_serv_op.cc
index 698ff22997891bc31eae628fe8eaf2ce71238544..7294acc3e357de174faf7ca88099626d56148471 100644
--- a/paddle/fluid/operators/listen_and_serv_op.cc
+++ b/paddle/fluid/operators/listen_and_serv_op.cc
@@ -329,7 +329,8 @@ void ListenAndServOp::RunImpl(const framework::Scope &scope,
   // Write to a file of server selected port for python use.
   SavePort();
   if (sync_mode) {
-    RunSyncLoop(&executor, program, &recv_scope, prefetch_block_id_list);
+    RunSyncLoop(&executor, program, &recv_scope, prefetch_block_id_list,
+                checkpoint_point_block_id);
   } else {
     RunAsyncLoop(&executor, program);
   }