diff --git a/custom_components/alfen_wallbox/__init__.py b/custom_components/alfen_wallbox/__init__.py
index 2e4df14..bd81ebe 100644
--- a/custom_components/alfen_wallbox/__init__.py
+++ b/custom_components/alfen_wallbox/__init__.py
@@ -1,10 +1,10 @@
 """Alfen Wallbox integration."""
 
 import asyncio
-from asyncio import timeout
 import logging
 
 from aiohttp import ClientConnectionError
+from async_timeout import timeout
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
diff --git a/custom_components/alfen_wallbox/config_flow.py b/custom_components/alfen_wallbox/config_flow.py
index a702a15..167efb6 100644
--- a/custom_components/alfen_wallbox/config_flow.py
+++ b/custom_components/alfen_wallbox/config_flow.py
@@ -1,9 +1,9 @@
 """Config flow for the Alfen Wallbox platform."""
 import asyncio
-from asyncio import timeout
 import logging
 
 from aiohttp import ClientError
+from async_timeout import timeout
 import voluptuous as vol
 
 from homeassistant import config_entries