diff --git a/tests/test_connection.py b/tests/test_connection.py
index 1fd776c..3b004c5 100644
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -30,6 +30,7 @@ from aioesphomeapi.core import (
     InvalidAuthAPIError,
     RequiresEncryptionAPIError,
     ResolveAPIError,
+    SocketClosedAPIError,
     TimeoutAPIError,
 )
 
@@ -461,6 +462,7 @@ async def test_finish_connection_times_out(
     [
         (OSError("Socket error"), HandshakeAPIError),
         (APIConnectionError, APIConnectionError),
+        (SocketClosedAPIError, SocketClosedAPIError),
         (asyncio.TimeoutError, TimeoutAPIError),
         (asyncio.CancelledError, APIConnectionError),
     ],