diff --git a/tests/test_aiohttp_websocket_exceptions.py b/tests/test_aiohttp_websocket_exceptions.py
index d50ac887..48312544 100644
--- a/tests/test_aiohttp_websocket_exceptions.py
+++ b/tests/test_aiohttp_websocket_exceptions.py
@@ -310,11 +310,6 @@ async def test_aiohttp_websocket_server_closing_after_ack(
 
     query = gql("query { hello }")
 
-    with pytest.raises(ConnectionResetError):
-        await session.execute(query)
-
-    await session.transport.wait_closed()
-
     with pytest.raises(TransportClosed):
         await session.execute(query)
 
diff --git a/tests/test_aiohttp_websocket_query.py b/tests/test_aiohttp_websocket_query.py
index 3062e5e7..2899038e 100644
--- a/tests/test_aiohttp_websocket_query.py
+++ b/tests/test_aiohttp_websocket_query.py
@@ -245,7 +245,7 @@ async def test_aiohttp_websocket_server_closing_after_first_query(
 
     # Now the server is closed but we don't know it yet, we have to send a query
     # to notice it and to receive the exception
-    with pytest.raises(ConnectionResetError):
+    with pytest.raises(TransportClosed):
         await session.execute(query)