fix value error for stations where state value is not found

pantherale0 · Jun 20, 2024 · c0d2b50 · c0d2b50
1 parent f9208b6
commit c0d2b50
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/fuel_prices/sensor.py b/custom_components/fuel_prices/sensor.py
@@ -94,13 +94,13 @@ def name(self) -> str:
     @property
     def native_unit_of_measurement(self) -> str:
         """Return unit of measurement."""
-        if self.state_value == "name":
+        if isinstance(self.native_value, str):
             return None
         return self._fuel_station.currency.upper()
 
     @property
     def state_class(self) -> str:
         """Return state type."""
-        if self.state_value == "name":
+        if isinstance(self.native_value, str):
             return None
         return "total"