diff --git a/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java b/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java
index 1a89f691..1ba3730b 100644
--- a/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java
+++ b/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java
@@ -157,7 +157,7 @@ public class WebsocketConsumerTests {
 	private WebSocketSession doHandshake(WebsocketConsumerClientHandler handler)
 			throws InterruptedException, ExecutionException {
 		String wsEndpoint = "ws://localhost:" + this.consumerServer.getPort() + this.properties.getPath();
-		return new StandardWebSocketClient().doHandshake(handler, wsEndpoint).get();
+		return new StandardWebSocketClient().execute(handler, wsEndpoint).get();
 	}
 
 	private List<String> submitMultipleMessages(int messageCount) {