diff --git a/src/util/hass-attributes-util.ts b/src/util/hass-attributes-util.ts
index feaf4c28fb..0d08b6ed87 100644
--- a/src/util/hass-attributes-util.ts
+++ b/src/util/hass-attributes-util.ts
@@ -124,7 +124,7 @@ hassAttributeUtil.LOGIC_STATE_ATTRIBUTES = {
   },
   state_class: {
     type: "array",
-    options: { sensor: ["measurement"] },
+    options: { sensor: ["measurement", "total", "total_increasing"] },
     description: "State class",
     domains: ["sensor"],
   },