diff --git a/framework-docs/src/main/java/org/springframework/docs/web/websocket/stomp/websocketstomphandlebrokerrelayconfigure/WebSocketConfiguration.java b/framework-docs/src/main/java/org/springframework/docs/web/websocket/stomp/websocketstomphandlebrokerrelayconfigure/WebSocketConfiguration.java
index 926da7d780..eb1b255157 100644
--- a/framework-docs/src/main/java/org/springframework/docs/web/websocket/stomp/websocketstomphandlebrokerrelayconfigure/WebSocketConfiguration.java
+++ b/framework-docs/src/main/java/org/springframework/docs/web/websocket/stomp/websocketstomphandlebrokerrelayconfigure/WebSocketConfiguration.java
@@ -23,7 +23,6 @@ import org.springframework.messaging.simp.config.MessageBrokerRegistry;
 import org.springframework.messaging.simp.stomp.StompReactorNettyCodec;
 import org.springframework.messaging.tcp.reactor.ReactorNettyTcpClient;
 import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
-import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
 import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
 
 // tag::snippet[]
@@ -41,7 +40,7 @@ public class WebSocketConfiguration implements WebSocketMessageBrokerConfigurer
 
 	private ReactorNettyTcpClient<byte[]> createTcpClient() {
 		return new ReactorNettyTcpClient<>(
-				client -> client.addressSupplier(() -> new InetSocketAddress(0)),
+				client -> client.remoteAddress(() -> new InetSocketAddress(0)),
 				new StompReactorNettyCodec());
 	}
 }