diff --git a/src/common/entity/compute_state_display.ts b/src/common/entity/compute_state_display.ts
index bd61083c82..cf69669ac6 100644
--- a/src/common/entity/compute_state_display.ts
+++ b/src/common/entity/compute_state_display.ts
@@ -165,6 +165,7 @@ export const computeStateDisplayFromEntityAttributes = (
   // state is a timestamp
   if (
     [
+      "ai_task",
       "button",
       "conversation",
       "event",