bug fix

totalitarian · Oct 6, 2021 · 697503d · 697503d
1 parent 83f0d3a
commit 697503d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/custom_components/hildebrandglow_dcc/sensor.py b/custom_components/hildebrandglow_dcc/sensor.py
@@ -192,11 +192,10 @@ def device_info(self) -> Optional[Dict[str, Any]]:
     @property
     def device_class(self) -> str:
         """Return the device class."""
-        if self._state is not None and self._state["units"] == "kWh":
-            if self.resource["dataSourceResourceTypeInfo"]["type"] == "ELEC":
-                return DEVICE_CLASS_ENERGY
-            elif self.resource["dataSourceResourceTypeInfo"]["type"] == "GAS":
-                return DEVICE_CLASS_GAS
+        if self.resource["dataSourceResourceTypeInfo"]["type"] == "ELEC":
+            return DEVICE_CLASS_ENERGY
+        if self.resource["dataSourceResourceTypeInfo"]["type"] == "GAS":
+            return DEVICE_CLASS_GAS
         if self._state is not None and self._state["units"] == "pence":
             return DEVICE_CLASS_MONETARY
         return None