diff --git a/homeassistant/components/modern_forms/sensor.py b/homeassistant/components/modern_forms/sensor.py index 5c9e0a18575..6d2ef5b6dab 100644 --- a/homeassistant/components/modern_forms/sensor.py +++ b/homeassistant/components/modern_forms/sensor.py @@ -3,9 +3,8 @@ from __future__ import annotations from datetime import datetime -from homeassistant.components.sensor import SensorEntity +from homeassistant.components.sensor import SensorDeviceClass, SensorEntity from homeassistant.config_entries import ConfigEntry -from homeassistant.const import DEVICE_CLASS_TIMESTAMP from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import StateType @@ -70,7 +69,7 @@ class ModernFormsLightTimerRemainingTimeSensor(ModernFormsSensor): key="light_timer_remaining_time", name=f"{coordinator.data.info.device_name} Light Sleep Time", ) - self._attr_device_class = DEVICE_CLASS_TIMESTAMP + self._attr_device_class = SensorDeviceClass.TIMESTAMP @property def native_value(self) -> StateType | datetime: @@ -100,7 +99,7 @@ class ModernFormsFanTimerRemainingTimeSensor(ModernFormsSensor): key="fan_timer_remaining_time", name=f"{coordinator.data.info.device_name} Fan Sleep Time", ) - self._attr_device_class = DEVICE_CLASS_TIMESTAMP + self._attr_device_class = SensorDeviceClass.TIMESTAMP @property def native_value(self) -> StateType | datetime: