diff --git a/custom_components/ha_text_ai/sensor.py b/custom_components/ha_text_ai/sensor.py
index f036b37..de6c982 100644
--- a/custom_components/ha_text_ai/sensor.py
+++ b/custom_components/ha_text_ai/sensor.py
@@ -59,13 +59,10 @@ async def async_setup_entry(
     coordinator = hass.data[DOMAIN][entry.entry_id]
     async_add_entities([HATextAISensor(coordinator, entry)], True)
 
-from homeassistant.components.sensor import SensorDeviceClass
-
 class HATextAISensor(CoordinatorEntity, SensorEntity):
     """HA text AI Sensor."""
 
     _attr_has_entity_name = True
-    _attr_device_class = SensorDeviceClass.TEXT  # Оставляем как есть
 
     def __init__(
         self,