diff --git a/spring-boot/src/main/java/org/baeldung/websocket/client/StompClient.java b/spring-boot/src/main/java/org/baeldung/websocket/client/StompClient.java
index da2dbeac3d..71f5471fb1 100644
--- a/spring-boot/src/main/java/org/baeldung/websocket/client/StompClient.java
+++ b/spring-boot/src/main/java/org/baeldung/websocket/client/StompClient.java
@@ -1,20 +1,15 @@
 package org.baeldung.websocket.client;
 
-import java.util.Collections;
-import java.util.List;
 import java.util.Scanner;
 
 import org.springframework.messaging.converter.MappingJackson2MessageConverter;
 import org.springframework.messaging.simp.stomp.StompSessionHandler;
+import org.springframework.web.socket.client.WebSocketClient;
 import org.springframework.web.socket.client.standard.StandardWebSocketClient;
 import org.springframework.web.socket.messaging.WebSocketStompClient;
-import org.springframework.web.socket.sockjs.client.SockJsClient;
-import org.springframework.web.socket.sockjs.client.Transport;
-import org.springframework.web.socket.sockjs.client.WebSocketTransport;
 
 /**
  * Stand alone WebSocketStompClient.
- * @author Kalyan
  *
  */
 public class StompClient {
@@ -22,13 +17,11 @@ public class StompClient {
     private static String URL = "ws://localhost:8080/spring-mvc-java/chat";
 
     public static void main(String[] args) {
-        Transport webSocketTransport = new WebSocketTransport(new StandardWebSocketClient());
-        List<Transport> transports = Collections.singletonList(webSocketTransport);
-        SockJsClient sockJsClient = new SockJsClient(transports);
-
-        WebSocketStompClient stompClient = new WebSocketStompClient(sockJsClient);
+        WebSocketClient client = new StandardWebSocketClient();
+        WebSocketStompClient stompClient = new WebSocketStompClient(client);
+        
         stompClient.setMessageConverter(new MappingJackson2MessageConverter());
-
+        
         StompSessionHandler sessionHandler = new MyStompSessionHandler();
         stompClient.connect(URL, sessionHandler);