🔧 fix device_class for submersion sensor (#145)

al-one · Jul 25, 2021 · 4f5cc31 · 4f5cc31
1 parent 15fc930
commit 4f5cc31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/xiaomi_miot/binary_sensor.py b/custom_components/xiaomi_miot/binary_sensor.py
@@ -11,7 +11,7 @@
     BinarySensorEntity,
     DEVICE_CLASS_MOTION,
     DEVICE_CLASS_DOOR,
-    DEVICE_CLASS_SAFETY,
+    DEVICE_CLASS_MOISTURE,
 )
 
 from . import (
@@ -101,7 +101,7 @@ def __init__(self, config, miot_service: MiotService, **kwargs):
 
         if miot_service.name in ['submersion_sensor']:
             self._prop_state = miot_service.get_property('submersion_state') or self._prop_state
-            self._vars['device_class'] = DEVICE_CLASS_SAFETY
+            self._vars['device_class'] = DEVICE_CLASS_MOISTURE
 
         self._state_attrs.update({
             'entity_class': self.__class__.__name__,