diff --git a/src/treaty2/ws.ts b/src/treaty2/ws.ts
index 75da87a..903defe 100644
--- a/src/treaty2/ws.ts
+++ b/src/treaty2/ws.ts
@@ -25,7 +25,7 @@ export class EdenWS<in out Schema extends InputSchema<any> = {}> {
 
     on<K extends keyof WebSocketEventMap>(
         type: K,
-        listener: (event: Treaty.WSEvent<K, Schema['response']>) => void,
+        listener: (event: Treaty.WSEvent<K, Schema['response'][200]>) => void,
         options?: boolean | AddEventListenerOptions
     ) {
         return this.addEventListener(type, listener, options)
@@ -43,7 +43,7 @@ export class EdenWS<in out Schema extends InputSchema<any> = {}> {
 
     subscribe(
         onMessage: (
-            event: Treaty.WSEvent<'message', Schema['response']>
+            event: Treaty.WSEvent<'message', Schema['response'][200]>
         ) => void,
         options?: boolean | AddEventListenerOptions
     ) {
@@ -52,7 +52,7 @@ export class EdenWS<in out Schema extends InputSchema<any> = {}> {
 
     addEventListener<K extends keyof WebSocketEventMap>(
         type: K,
-        listener: (event: Treaty.WSEvent<K, Schema['response']>) => void,
+        listener: (event: Treaty.WSEvent<K, Schema['response'][200]>) => void,
         options?: boolean | AddEventListenerOptions
     ) {
         this.ws.addEventListener(