diff --git a/custom_components/ha_text_ai/sensor.py b/custom_components/ha_text_ai/sensor.py
index 56758f4..0808764 100644
--- a/custom_components/ha_text_ai/sensor.py
+++ b/custom_components/ha_text_ai/sensor.py
@@ -100,6 +100,14 @@ class HATextAISensor(CoordinatorEntity, SensorEntity):
             return dt_util.as_local(last_response['timestamp'])
         return self._current_state
 
+    @property
+    def state(self) -> StateType:
+        """Return the state of the sensor."""
+        last_response = self.coordinator.last_response
+        if last_response and 'timestamp' in last_response:
+            return dt_util.as_local(last_response['timestamp'])
+        return self._current_state
+
     @property
     def extra_state_attributes(self) -> Dict[str, Any]:
         """Return entity specific state attributes."""