Use native_unit_of_measurement to prevent recursion. (#209)

Co-authored-by: Felix Stürmer <[email protected]>
weltenwort · May 8, 2022 · 9e1202d · 9e1202d
1 parent 436df7d
commit 9e1202d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/rct_power/lib/entity.py b/custom_components/rct_power/lib/entity.py
@@ -130,8 +130,8 @@ def device_class(self):
         if device_class := super().device_class:
             return device_class
 
-        if self.unit_of_measurement:
-            return guess_device_class_from_unit(self.unit_of_measurement)
+        if self.native_unit_of_measurement:
+            return guess_device_class_from_unit(self.native_unit_of_measurement)
 
         return None