1 files changed, 2 insertions, 5 deletions
diff --git a/src/mongo/transport/grpc/mock_server_context.h b/src/mongo/transport/grpc/mock_server_context.h
index bd1fa05d47b..0c867f43e18 100644
--- a/src/mongo/transport/grpc/mock_server_context.h
+++ b/src/mongo/transport/grpc/mock_server_context.h
@@ -43,11 +43,8 @@ public:
     const MetadataView& getClientMetadata() const override;
     Date_t getDeadline() const override;
     bool isCancelled() const override;
-    HostAndPort getHostAndPort() const override {
-        return _stream->_hostAndPort;
-    }
-    CancellationToken getCancellationToken() {
-        return _stream->_cancellationSource.token();
+    HostAndPort getRemote() const override {
+        return _stream->_remote;
     }
     void tryCancel() override;