1 files changed, 6 insertions, 2 deletions
diff --git a/pw_rpc/raw/public/pw_rpc/raw/server_reader_writer.h b/pw_rpc/raw/public/pw_rpc/raw/server_reader_writer.h
index 4de87fc03..91a85e552 100644
--- a/pw_rpc/raw/public/pw_rpc/raw/server_reader_writer.h
+++ b/pw_rpc/raw/public/pw_rpc/raw/server_reader_writer.h
@@ -75,7 +75,8 @@ class RawServerReaderWriter : private internal::ServerCall {
   // Functions for setting the callbacks.
   using internal::Call::set_on_error;
   using internal::Call::set_on_next;
-  using internal::ServerCall::set_on_client_stream_end;
+  using internal::ServerCall::set_on_completion_requested;
+  using internal::ServerCall::set_on_completion_requested_if_enabled;
 
   // Sends a response packet with the given raw payload.
   using internal::Call::Write;
@@ -140,7 +141,8 @@ class RawServerReader : private RawServerReaderWriter {
   using RawServerReaderWriter::active;
   using RawServerReaderWriter::channel_id;
 
-  using RawServerReaderWriter::set_on_client_stream_end;
+  using RawServerReaderWriter::set_on_completion_requested;
+  using RawServerReaderWriter::set_on_completion_requested_if_enabled;
   using RawServerReaderWriter::set_on_error;
   using RawServerReaderWriter::set_on_next;
 
@@ -188,6 +190,8 @@ class RawServerWriter : private RawServerReaderWriter {
   using RawServerReaderWriter::active;
   using RawServerReaderWriter::channel_id;
 
+  using RawServerReaderWriter::set_on_completion_requested;
+  using RawServerReaderWriter::set_on_completion_requested_if_enabled;
   using RawServerReaderWriter::set_on_error;
 
   using RawServerReaderWriter::Finish;