diff --git a/custom_components/pstryk/mqtt_publisher.py b/custom_components/pstryk/mqtt_publisher.py
index e4018e3..08dd38a 100644
--- a/custom_components/pstryk/mqtt_publisher.py
+++ b/custom_components/pstryk/mqtt_publisher.py
@@ -1,9 +1,8 @@
 """MQTT Publisher for Pstryk Energy integration."""
 import logging
 import json
-from datetime import datetime, timedelta
+from datetime import timedelta
 import asyncio
-from homeassistant.helpers.entity import Entity
 from homeassistant.core import HomeAssistant
 from homeassistant.util import dt as dt_util
 from homeassistant.components import mqtt