diff --git a/spring-cloud-gateway-core/src/test/java/org/springframework/cloud/gateway/test/websocket/WebSocketIntegrationTests.java b/spring-cloud-gateway-core/src/test/java/org/springframework/cloud/gateway/test/websocket/WebSocketIntegrationTests.java
index a51e7e31..f9c1bae7 100644
--- a/spring-cloud-gateway-core/src/test/java/org/springframework/cloud/gateway/test/websocket/WebSocketIntegrationTests.java
+++ b/spring-cloud-gateway-core/src/test/java/org/springframework/cloud/gateway/test/websocket/WebSocketIntegrationTests.java
@@ -196,7 +196,7 @@ public class WebSocketIntegrationTests {
 				session -> {
 					logger.debug("Starting to send messages");
 					return session
-							.send(input.doOnNext(s -> logger.debug("outbound " + s)).map(session::textMessage))
+							.send(input.doOnNext(s -> logger.debug("outbound " + s)).map(s -> session.textMessage(s)))
 							.thenMany(session.receive().take(count).map(WebSocketMessage::getPayloadAsText))
 							.subscribeWith(output)
 							.doOnNext(s -> logger.debug("inbound " + s))
@@ -220,7 +220,7 @@ public class WebSocketIntegrationTests {
 				session -> {
 					logger.debug("Starting to send messages");
 					return session
-							.send(input.doOnNext(s -> logger.debug("outbound " + s)).map(session::textMessage))
+							.send(input.doOnNext(s -> logger.debug("outbound " + s)).map(s -> session.textMessage(s)))
 							.thenMany(session.receive().take(count).map(WebSocketMessage::getPayloadAsText))
 							.subscribeWith(output)
 							.doOnNext(s -> logger.debug("inbound " + s))