1 files changed, 2 insertions, 2 deletions
diff --git a/swiftstory/client.py b/swiftstory/client.py
index cd5e82a..0831cb0 100644
--- a/swiftstory/client.py
+++ b/swiftstory/client.py
@@ -5,7 +5,7 @@ import logging
 from typing import Optional
 
 import websockets.exceptions
-from websockets.server import WebSocketServerProtocol
+from websockets.asyncio.server import ServerConnection
 
 from swiftstory.game import Game
 from swiftstory.game_manager import GameManager
@@ -17,7 +17,7 @@ class Client:
     """ Represent a client.
     A client manages a (web)socket to communicate with the outside world.
     It also manages the associated player when in a game. """
-    def __init__(self, socket: WebSocketServerProtocol, game_manager: GameManager) -> None:
+    def __init__(self, socket: ServerConnection, game_manager: GameManager) -> None:
         self.game: Optional[Game] = None
         self.game_manager: GameManager = game_manager