1 files changed, 5 insertions, 6 deletions
diff --git a/swiftstory/__main__.py b/swiftstory/__main__.py
index 3e2bcff..453f25f 100644
--- a/swiftstory/__main__.py
+++ b/swiftstory/__main__.py
@@ -6,13 +6,13 @@ import argparse
 import asyncio
 import logging
 
-from websockets.server import serve
+from websockets.asyncio.server import serve
 
 from swiftstory.game_manager import GameManager
 from swiftstory.interface.ws import WebsocketsInterface
 
 
-def main() -> None:
+async def main() -> None:
     """ Entry point: we create the game manager and start the Websockets
     server. """
     parser = argparse.ArgumentParser()
@@ -23,12 +23,11 @@ def main() -> None:
     logging.basicConfig(level=logging.INFO)
 
     ws_interface = WebsocketsInterface(GameManager())
-    start_server = serve(ws_interface.connection_handler,
+    server = await serve(ws_interface.connection_handler,
                          args['listen'], args['port'])
 
-    asyncio.get_event_loop().run_until_complete(start_server)
-    asyncio.get_event_loop().run_forever()
+    await server.serve_forever()
 
 
 if __name__ == '__main__':
-    main()
+    asyncio.run(main())