diff --git a/aiomysql/connection.py b/aiomysql/connection.py index 32da048c..8bf5cbf4 100644 --- a/aiomysql/connection.py +++ b/aiomysql/connection.py @@ -596,7 +596,7 @@ async def _read_packet(self, packet_type=MysqlPacket): async def _read_bytes(self, num_bytes): try: data = await self._reader.readexactly(num_bytes) - except asyncio.streams.IncompleteReadError as e: + except asyncio.IncompleteReadError as e: msg = "Lost connection to MySQL server during query" raise OperationalError(2013, msg) from e except (IOError, OSError) as e: