diff --git a/tests/test_timeout.py b/tests/test_timeout.py
index <HASH>..<HASH> 100644
--- a/tests/test_timeout.py
+++ b/tests/test_timeout.py
@@ -11,10 +11,10 @@ SENSOR_ID = 1
 async def test_timeout(httpx_mock: HTTPXMock):
     """Test if the connection is hitting the timeout."""
 
-    def raise_timeout(request, extensions: dict):
+    def raise_timeout(request):
         """Set the timeout for the requests."""
         raise httpx.ReadTimeout(
-            f"Unable to read within {extensions['timeout']}", request=request
+            f"Unable to read within {request.extensions['timeout']}", request=request
         )
 
     httpx_mock.add_callback(raise_timeout)
Fix for later pytest-httpx releases