diff --git a/rust/perspective-python/perspective/handlers/aiohttp.py b/rust/perspective-python/perspective/handlers/aiohttp.py
index d1685769e8..421889720d 100644
--- a/rust/perspective-python/perspective/handlers/aiohttp.py
+++ b/rust/perspective-python/perspective/handlers/aiohttp.py
@@ -39,7 +39,7 @@ def __init__(self, **kwargs):
         self._request = kwargs.pop("request")
         super().__init__(**kwargs)
 
-    async def run(self) -> None:
+    async def run(self) -> web.WebSocketResponse:
         def inner(msg):
             asyncio.get_running_loop().create_task(self._ws.send_bytes(msg))
 
@@ -53,3 +53,4 @@ def inner(msg):
 
         finally:
             self.session.close()
+        return self._ws