diff --git a/types/koa-websocket/index.d.ts b/types/koa-websocket/index.d.ts
index e7a6c39a7d..ebbfeee8f2 100644
--- a/types/koa-websocket/index.d.ts
+++ b/types/koa-websocket/index.d.ts
@@ -1,6 +1,6 @@
-// Type definitions for koa-websocket 2.1
+// Type definitions for koa-websocket 5.0
 // Project: https://github.com/kudos/koa-websocket
-// Definitions by: My Self <https://github.com/me>
+// Definitions by: Maël Lavault <https://github.com/moimael>
 // Definitions: https://github.com/DefinitelyTyped/DefinitelyTyped
 // TypeScript Version: 2.3
 
@@ -21,7 +21,7 @@ declare class KoaWebsocketServer {
     middleware: Koa.Middleware[];
 
     constructor(app: Koa);
-    listen(server: http.Server | https.Server): ws.Server;
+    listen(options: ws.ServerOptions): ws.Server;
     onConnection(handler: KoaWebsocketConnectionHandler): void;
     use(middleware: KoaWebsocketMiddleware): this;
 }