diff --git a/custom_components/reos_integration/__init__.py b/custom_components/reos_integration/__init__.py index 2968f5c..32543fe 100755 --- a/custom_components/reos_integration/__init__.py +++ b/custom_components/reos_integration/__init__.py @@ -1,5 +1,6 @@ from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant +from homeassistant.const import Platform import logging from .reos_api import ReosApi @@ -46,7 +47,5 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: hass.data[DOMAIN][username]["coordinator"] = coordinator await coordinator.async_config_entry_first_refresh() - await hass.config_entries.async_forward_entry_setup(entry, "button") - await hass.config_entries.async_forward_entry_setup(entry, "sensor") - await hass.config_entries.async_forward_entry_setup(entry, "binary_sensor") + await hass.config_entries.async_forward_entry_setups(entry, [Platform.BUTTON, Platform.SENSOR, Platform.BINARY_SENSOR]) return True