diff --git a/kotlin-sdk-client/build.gradle.kts b/kotlin-sdk-client/build.gradle.kts
index 7db177cd..2466f8ef 100644
--- a/kotlin-sdk-client/build.gradle.kts
+++ b/kotlin-sdk-client/build.gradle.kts
@@ -43,6 +43,7 @@ kotlin {
                 implementation(kotlin("test"))
                 implementation(dependencies.platform(libs.ktor.bom))
                 implementation(libs.ktor.client.mock)
+                implementation(libs.ktor.server.websockets)
                 implementation(libs.kotlinx.coroutines.test)
                 implementation(libs.ktor.client.logging)
             }
diff --git a/kotlin-sdk-core/build.gradle.kts b/kotlin-sdk-core/build.gradle.kts
index 68e56185..a3e1b58c 100644
--- a/kotlin-sdk-core/build.gradle.kts
+++ b/kotlin-sdk-core/build.gradle.kts
@@ -113,8 +113,8 @@ kotlin {
                 api(libs.kotlinx.serialization.json)
                 api(libs.kotlinx.coroutines.core)
                 api(libs.kotlinx.io.core)
-                api(libs.ktor.server.websockets)
                 api(libs.kotlinx.collections.immutable)
+                implementation(libs.ktor.server.websockets)
                 implementation(libs.kotlin.logging)
             }
         }
diff --git a/kotlin-sdk-server/build.gradle.kts b/kotlin-sdk-server/build.gradle.kts
index c5feae0a..64a27497 100644
--- a/kotlin-sdk-server/build.gradle.kts
+++ b/kotlin-sdk-server/build.gradle.kts
@@ -12,6 +12,7 @@ kotlin {
                 api(project(":kotlin-sdk-core"))
                 api(libs.ktor.server.core)
                 api(libs.ktor.server.sse)
+                implementation(libs.ktor.server.websockets)
                 implementation(libs.kotlin.logging)
             }
         }
diff --git a/kotlin-sdk-test/build.gradle.kts b/kotlin-sdk-test/build.gradle.kts
index b51398b3..c7f8b60d 100644
--- a/kotlin-sdk-test/build.gradle.kts
+++ b/kotlin-sdk-test/build.gradle.kts
@@ -18,6 +18,7 @@ kotlin {
                 implementation(libs.kotlinx.coroutines.test)
                 implementation(libs.ktor.client.cio)
                 implementation(libs.ktor.server.cio)
+                implementation(libs.ktor.server.websockets)
                 implementation(libs.ktor.server.test.host)
             }
         }