diff --git a/sensor.py b/sensor.py
index 21ba3bb..8b19bfb 100644
--- a/sensor.py
+++ b/sensor.py
@@ -1,5 +1,4 @@
-import datetime
-from homeassistant.components.switch import SensorEntity
+from homeassistant.components.sensor import SensorEntity
 from homeassistant.core import HomeAssistant
 from homeassistant.const import TIME_MINUTES
 from homeassistant.helpers.entity_platform import AddEntitiesCallback