diff --git a/homeassistant/components/hvv_departures/binary_sensor.py b/homeassistant/components/hvv_departures/binary_sensor.py index 106279cbb91..2e47698ecd7 100644 --- a/homeassistant/components/hvv_departures/binary_sensor.py +++ b/homeassistant/components/hvv_departures/binary_sensor.py @@ -15,6 +15,7 @@ from homeassistant.components.binary_sensor import ( ) from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant +from homeassistant.helpers.device_registry import DeviceEntryType from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.update_coordinator import ( @@ -150,6 +151,7 @@ class HvvDepartureBinarySensor(CoordinatorEntity, BinarySensorEntity): def device_info(self): """Return the device info for this sensor.""" return DeviceInfo( + entry_type=DeviceEntryType.SERVICE, identifiers={ ( DOMAIN, diff --git a/homeassistant/components/hvv_departures/sensor.py b/homeassistant/components/hvv_departures/sensor.py index e2de1758dd5..3a44bc59c72 100644 --- a/homeassistant/components/hvv_departures/sensor.py +++ b/homeassistant/components/hvv_departures/sensor.py @@ -11,6 +11,7 @@ from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_ID from homeassistant.core import HomeAssistant from homeassistant.helpers import aiohttp_client +from homeassistant.helpers.device_registry import DeviceEntryType from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.util import Throttle @@ -167,6 +168,7 @@ class HVVDepartureSensor(SensorEntity): def device_info(self): """Return the device info for this sensor.""" return DeviceInfo( + entry_type=DeviceEntryType.SERVICE, identifiers={ ( DOMAIN,