diff --git a/homeassistant/components/unifi/sensor.py b/homeassistant/components/unifi/sensor.py index 28db9abb94f..a0cd3a7f1e7 100644 --- a/homeassistant/components/unifi/sensor.py +++ b/homeassistant/components/unifi/sensor.py @@ -430,10 +430,10 @@ class UnifiSensorEntity(UnifiEntity[HandlerT, ApiItemT], SensorEntity): def _make_disconnected(self, *_: core_Event) -> None: """No heart beat by device. - Reset sensor value to 0 when client device is disconnected + Set sensor as unavailable when client device is disconnected """ - if self._attr_native_value != 0: - self._attr_native_value = 0 + if self._attr_available: + self._attr_available = False self.async_write_ha_state() @callback diff --git a/tests/components/unifi/test_sensor.py b/tests/components/unifi/test_sensor.py index 1a3c81ec4c4..9ebdd207b54 100644 --- a/tests/components/unifi/test_sensor.py +++ b/tests/components/unifi/test_sensor.py @@ -416,8 +416,8 @@ async def test_bandwidth_sensors( async_fire_time_changed(hass, new_time) await hass.async_block_till_done() - assert hass.states.get("sensor.wireless_client_rx").state == "0" - assert hass.states.get("sensor.wireless_client_tx").state == "0" + assert hass.states.get("sensor.wireless_client_rx").state == STATE_UNAVAILABLE + assert hass.states.get("sensor.wireless_client_tx").state == STATE_UNAVAILABLE # Disable option