diff --git a/packages/widgets/src/smart-home/entity-state/component.tsx b/packages/widgets/src/smart-home/entity-state/component.tsx
index 587978a42..f11cfa844 100644
--- a/packages/widgets/src/smart-home/entity-state/component.tsx
+++ b/packages/widgets/src/smart-home/entity-state/component.tsx
@@ -76,6 +76,7 @@ export default function SmartHomeEntityStateWidget({
 
   return (
     <UnstyledButton
+      mod={{ "entity-state": entityState, "entity-id": options.entityId }}
       onClick={handleClick}
       w="100%"
       h="100%"