diff --git a/custom_components/pyscript/__init__.py b/custom_components/pyscript/__init__.py
index 6b177aa..c0f0137 100644
--- a/custom_components/pyscript/__init__.py
+++ b/custom_components/pyscript/__init__.py
@@ -22,7 +22,7 @@
     SERVICE_RELOAD,
 )
 from homeassistant.core import Event as HAEvent, HomeAssistant, ServiceCall
-from homeassistant.core_config import Config
+from homeassistant.helpers.typing import ConfigType
 from homeassistant.exceptions import HomeAssistantError
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.restore_state import DATA_RESTORE_STATE
@@ -65,7 +65,7 @@
 CONFIG_SCHEMA = vol.Schema({DOMAIN: PYSCRIPT_SCHEMA}, extra=vol.ALLOW_EXTRA)
 
 
-async def async_setup(hass: HomeAssistant, config: Config) -> bool:
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
     """Component setup, run import config flow for each entry in config."""
     await restore_state(hass)
     if DOMAIN in config: