From f1f05cdf1b7a75b7f76d09e43c1749fa17c8768d Mon Sep 17 00:00:00 2001
From: Maciej Bieniek <bieniu@users.noreply.github.com>
Date: Tue, 17 Aug 2021 11:57:45 +0200
Subject: [PATCH] Use DEVICE_CLASS_UPDATE in Shelly integration (#54746)

---
 homeassistant/components/shelly/binary_sensor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/homeassistant/components/shelly/binary_sensor.py b/homeassistant/components/shelly/binary_sensor.py
index dd1b3a9d66d..96d62152830 100644
--- a/homeassistant/components/shelly/binary_sensor.py
+++ b/homeassistant/components/shelly/binary_sensor.py
@@ -12,6 +12,7 @@ from homeassistant.components.binary_sensor import (
     DEVICE_CLASS_POWER,
     DEVICE_CLASS_PROBLEM,
     DEVICE_CLASS_SMOKE,
+    DEVICE_CLASS_UPDATE,
     DEVICE_CLASS_VIBRATION,
     STATE_ON,
     BinarySensorEntity,
@@ -99,7 +100,7 @@ REST_SENSORS: Final = {
     ),
     "fwupdate": RestAttributeDescription(
         name="Firmware Update",
-        icon="mdi:update",
+        device_class=DEVICE_CLASS_UPDATE,
         value=lambda status, _: status["update"]["has_update"],
         default_enabled=False,
         extra_state_attributes=lambda status: {