1 files changed, 2 insertions, 2 deletions
diff --git a/swiftstory/interface/ws.py b/swiftstory/interface/ws.py
index 082ed1a..e9150dd 100644
--- a/swiftstory/interface/ws.py
+++ b/swiftstory/interface/ws.py
@@ -7,7 +7,7 @@ import logging
 from typing import Union
 
 import websockets.exceptions
-from websockets.server import WebSocketServerProtocol
+from websockets.asyncio.server import ServerConnection
 
 from swiftstory.client import Client
 from swiftstory.exception import WrongAction, JoinError
@@ -20,7 +20,7 @@ class WebsocketsInterface:
     def __init__(self, game_manager: GameManager):
         self.game_manager = game_manager
 
-    async def connection_handler(self, websocket: WebSocketServerProtocol, path: str) -> None:
+    async def connection_handler(self, websocket: ServerConnection) -> None:
         """ Coroutine that handles one websocket connection. """
         client = Client(websocket, self.game_manager)