diff --git a/custom_components/ocpp/__init__.py b/custom_components/ocpp/__init__.py
index fabf43f7..e43985ea 100644
--- a/custom_components/ocpp/__init__.py
+++ b/custom_components/ocpp/__init__.py
@@ -3,7 +3,8 @@
 import logging
 
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.core import Config, HomeAssistant
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.typing import ConfigType
 from homeassistant.helpers import device_registry
 import homeassistant.helpers.config_validation as cv
 import voluptuous as vol
@@ -50,7 +51,7 @@
 )
 
 
-async def async_setup(hass: HomeAssistant, config: Config):
+async def async_setup(hass: HomeAssistant, config: ConfigType):
     """Read configuration from yaml."""
 
     ocpp_config = config.get(DOMAIN, {})