diff --git a/test/test_protocol.py b/test/test_protocol.py
index 38c28ea..390582e 100644
--- a/test/test_protocol.py
+++ b/test/test_protocol.py
@@ -233,7 +233,7 @@ class Ei:
             while data:
                 self._data += data
                 data = self.sock.recv(1024)
-        except BlockingIOError:
+        except (BlockingIOError, ConnectionResetError):
             pass
         return self.data