diff --git a/src/main/java/com/olympus/apollo/config/WebSocketConfig.java b/src/main/java/com/olympus/apollo/config/WebSocketConfig.java
index 9353b00..3672c72 100644
--- a/src/main/java/com/olympus/apollo/config/WebSocketConfig.java
+++ b/src/main/java/com/olympus/apollo/config/WebSocketConfig.java
@@ -13,8 +13,8 @@ public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
     @Override
     public void registerStompEndpoints(StompEndpointRegistry registry) {
         registry.addEndpoint("/ws/endpoint")
-                .setAllowedOrigins("*");
-                //.withSockJS();
+                .setAllowedOrigins("http://olympus-api-gateway.olympusai.live")
+                .withSockJS();
     }
 
     @Override