diff --git a/build.gradle b/build.gradle
index aac83c3..3fc1875 100644
--- a/build.gradle
+++ b/build.gradle
@@ -18,9 +18,9 @@
     implementation 'org.springframework.boot:spring-boot-starter-web'
     providedRuntime 'org.springframework.boot:spring-boot-starter-tomcat'
     testImplementation 'org.springframework.boot:spring-boot-starter-test'
-    //providedCompile 'javax.websocket:javax.websocket-api:1.1'
 
-    //implementation 'org.springframework.boot:spring-boot-starter-websocket'
+    providedCompile 'javax.websocket:javax.websocket-api:1.1'
+    implementation 'org.springframework.boot:spring-boot-starter-websocket'
 }
 
 tasks.named('test') {