Fix bugs

Author: Bre77

custom_components/teslemetry/select.py

@@ -29,6 +29,7 @@
     TeslemetryVehicleEntity,
     TeslemetryEnergyInfoEntity,
     TeslemetryVehicleStreamEntity,
+    TeslemetryVehicleStreamSingleEntity,
 )
 from .models import TeslemetryEnergyData, TeslemetryVehicleData
 
@@ -218,7 +219,7 @@ async def async_added_to_hass(self) -> None:
 
         # Restore state
         if (state := await self.async_get_last_state()) is not None:
-            if state.state in self.entity_description.options:
+            if state.state in self._attr_options:
                 self._attr_current_option = state.state
 
         # Listen for streaming data
@@ -238,7 +239,7 @@ def _value_callback(self, value: int | None) -> None:
         if value is None:
             self._attr_current_option = None
         else:
-            self._attr_current_option = self.entity_description.options[value]
+            self._attr_current_option = self._attr_options[value]
         self.async_write_ha_state()
 
     def _climate_callback(self, value: bool | None) -> None: