diff --git a/homeassistant/components/flo/sensor.py b/homeassistant/components/flo/sensor.py index b2a0afdcb13..6d023731c86 100644 --- a/homeassistant/components/flo/sensor.py +++ b/homeassistant/components/flo/sensor.py @@ -12,6 +12,7 @@ from homeassistant.const import ( UnitOfPressure, UnitOfTemperature, UnitOfVolume, + UnitOfVolumeFlowRate, ) from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -98,8 +99,9 @@ class FloCurrentFlowRateSensor(FloEntity, SensorEntity): """Monitors the current water flow rate.""" _attr_icon = GAUGE_ICON - _attr_native_unit_of_measurement = "gpm" + _attr_native_unit_of_measurement = UnitOfVolumeFlowRate.GALLONS_PER_MINUTE _attr_state_class: SensorStateClass = SensorStateClass.MEASUREMENT + _attr_device_class = SensorDeviceClass.VOLUME_FLOW_RATE _attr_translation_key = "current_flow_rate" def __init__(self, device):