diff --git a/src/Layers/trpc/tests.ts b/src/Layers/trpc/tests.ts
index 359fb8b..e0ac6c0 100644
--- a/src/Layers/trpc/tests.ts
+++ b/src/Layers/trpc/tests.ts
@@ -1,4 +1,4 @@
-import { Config, Context, Effect, Layer } from "effect"
+import { Config, Effect, Layer } from "effect"
 import * as TRPC from "."
 import { ExpressApp, ExpressNodeHTTPServer } from "../express"
 
@@ -17,11 +17,16 @@ const router = TRPCBuilder.pipe(Effect.map(t => t.router({
 
 const { TRPCRouter, TRPCRouterLive } = TRPC.TRPCRouter.make(router)
 
+const { TRPCWebSocketServer, TRPCWebSocketServerLive } = TRPC.TRPCWebSocketServer.make(TRPCRouter)
+
 
 const ServerLive = Layer.empty.pipe(
     Layer.provideMerge(TRPC.TRPCExpressRoute.TRPCExpressRouteLive(TRPCRouter, {
         root: Config.succeed("/rpc")
     })),
+    Layer.provideMerge(TRPCWebSocketServerLive({
+        host: Config.succeed("/rpc")
+    })),
 
     Layer.provideMerge(TRPCRouterLive),
     Layer.provideMerge(TRPCBuilderLive),