diff --git a/homeassistant/components/shelly/sensor.py b/homeassistant/components/shelly/sensor.py
index 51cb462fa48..bd9041d3344 100644
--- a/homeassistant/components/shelly/sensor.py
+++ b/homeassistant/components/shelly/sensor.py
@@ -283,7 +283,7 @@ RPC_SENSORS: Final = {
         device_class=sensor.DEVICE_CLASS_TEMPERATURE,
         state_class=sensor.STATE_CLASS_MEASUREMENT,
         default_enabled=False,
-        entity_category=EntityCategory.DIAGNOSTIC,
+        entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
     ),
     "rssi": RpcAttributeDescription(
         key="wifi",