Fix base class

Teslemetry · Feb 4, 2024 · e117577 · e117577
1 parent 0249f3d
commit e117577
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/custom_components/teslemetry/switch.py b/custom_components/teslemetry/switch.py
@@ -19,6 +19,7 @@
 from .const import DOMAIN
 from .entity import (
     TeslemetryVehicleEntity,
+    TeslemetryEnergyInfoEntity
 )
 from .models import TeslemetryVehicleData, TeslemetryEnergyData
 
@@ -169,7 +170,7 @@ async def async_turn_off(self, **kwargs: Any) -> None:
         self.set((self.entity_description.key, False))
 
 
-class TeslemetryEnergyInfoSwitchEntity(TeslemetryVehicleEntity, SwitchEntity):
+class TeslemetryEnergyInfoSwitchEntity(TeslemetryEnergyInfoEntity, SwitchEntity):
     """Base class for Teslemetry Switch."""
 
     _attr_device_class = SwitchDeviceClass.SWITCH