diff --git a/custom_components/battery_sim/sensor.py b/custom_components/battery_sim/sensor.py
index 2b6529f..37dd489 100644
--- a/custom_components/battery_sim/sensor.py
+++ b/custom_components/battery_sim/sensor.py
@@ -8,6 +8,7 @@
     STATE_CLASS_MEASUREMENT,
     STATE_CLASS_TOTAL,
     SensorEntity,
+    ATTR_LAST_RESET
 )
 from homeassistant.const import (
     DEVICE_CLASS_ENERGY,