From 15feb430fca10c71d45e1a0bae3c8e305466f924 Mon Sep 17 00:00:00 2001
From: Michael <35783820+mib1185@users.noreply.github.com>
Date: Tue, 17 Aug 2021 19:15:02 +0200
Subject: [PATCH] Use DEVICE_CLASS_UPDATE in Synology DSM (#54769)

---
 homeassistant/components/synology_dsm/const.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/synology_dsm/const.py b/homeassistant/components/synology_dsm/const.py
index fdbbb5678c2..633c264f3c8 100644
--- a/homeassistant/components/synology_dsm/const.py
+++ b/homeassistant/components/synology_dsm/const.py
@@ -10,7 +10,10 @@ from synology_dsm.api.dsm.information import SynoDSMInformation
 from synology_dsm.api.storage.storage import SynoStorage
 from synology_dsm.api.surveillance_station import SynoSurveillanceStation
 
-from homeassistant.components.binary_sensor import DEVICE_CLASS_SAFETY
+from homeassistant.components.binary_sensor import (
+    DEVICE_CLASS_SAFETY,
+    DEVICE_CLASS_UPDATE,
+)
 from homeassistant.components.sensor import ATTR_STATE_CLASS, STATE_CLASS_MEASUREMENT
 from homeassistant.const import (
     ATTR_DEVICE_CLASS,
@@ -81,8 +84,8 @@ UPGRADE_BINARY_SENSORS: dict[str, EntityInfo] = {
     f"{SynoCoreUpgrade.API_KEY}:update_available": {
         ATTR_NAME: "Update available",
         ATTR_UNIT_OF_MEASUREMENT: None,
-        ATTR_ICON: "mdi:update",
-        ATTR_DEVICE_CLASS: None,
+        ATTR_ICON: None,
+        ATTR_DEVICE_CLASS: DEVICE_CLASS_UPDATE,
         ENTITY_ENABLE: True,
         ATTR_STATE_CLASS: None,
     },