diff --git a/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java b/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java
index e860f10d3146eb99eb3122fd7eea58dbef83a31c..08ee46096aa692e5fb840c66ff696c005fb0b79a 100644
--- a/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java
+++ b/spring-web-reactive/src/test/java/org/springframework/web/reactive/socket/server/WebSocketIntegrationTests.java
@@ -29,6 +29,7 @@ import org.springframework.context.annotation.Configuration;
 import org.springframework.web.reactive.HandlerMapping;
 import org.springframework.web.reactive.handler.SimpleUrlHandlerMapping;
 import org.springframework.web.reactive.socket.WebSocketHandler;
+import org.springframework.web.reactive.socket.WebSocketMessage;
 import org.springframework.web.reactive.socket.WebSocketSession;
 import org.springframework.web.reactive.socket.client.ReactorNettyWebSocketClient;
 import org.springframework.web.reactive.socket.client.RxNettyWebSocketClient;
@@ -68,13 +69,7 @@ public class WebSocketIntegrationTests extends AbstractWebSocketIntegrationTests
 		client.execute(getUrl("/echo"),
 				session -> session
 						.send(input.map(session::textMessage))
-						.thenMany(session.receive()
-								.take(count)
-								.map(message -> {
-									String text = message.getPayloadAsText();
-									message.release();
-									return text;
-								}))
+						.thenMany(session.receive().take(count).map(WebSocketMessage::getPayloadAsText))
 						.subscribeWith(output)
 						.then())
 				.blockMillis(5000);