diff --git a/homeassistant/components/zha/button.py b/homeassistant/components/zha/button.py index 6564f3bc39a..010c0f63e27 100644 --- a/homeassistant/components/zha/button.py +++ b/homeassistant/components/zha/button.py @@ -105,7 +105,7 @@ class ZHAIdentifyButton(ZHAButton): return None return cls(unique_id, zha_device, cluster_handlers, **kwargs) - _attr_device_class: ButtonDeviceClass = ButtonDeviceClass.UPDATE + _attr_device_class = ButtonDeviceClass.IDENTIFY _attr_entity_category = EntityCategory.DIAGNOSTIC _attr_name = "Identify" _command_name = "identify" diff --git a/tests/components/zha/test_button.py b/tests/components/zha/test_button.py index f3ee3d6dca7..2a2fbc92ace 100644 --- a/tests/components/zha/test_button.py +++ b/tests/components/zha/test_button.py @@ -143,7 +143,7 @@ async def test_button(hass: HomeAssistant, contact_sensor) -> None: state = hass.states.get(entity_id) assert state assert state.state == STATE_UNKNOWN - assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.UPDATE + assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.IDENTIFY entry = entity_registry.async_get(entity_id) assert entry @@ -168,7 +168,7 @@ async def test_button(hass: HomeAssistant, contact_sensor) -> None: state = hass.states.get(entity_id) assert state assert state.state == "2021-11-04T16:37:00+00:00" - assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.UPDATE + assert state.attributes[ATTR_DEVICE_CLASS] == ButtonDeviceClass.IDENTIFY async def test_frost_unlock(hass: HomeAssistant, tuya_water_valve) -> None: