diff --git a/src/button-card.ts b/src/button-card.ts
index fe8b1a3..c2e0794 100644
--- a/src/button-card.ts
+++ b/src/button-card.ts
@@ -1154,6 +1154,8 @@ class ButtonCard extends LitElement {
             ? html`
                 <ha-state-icon
                   .state=${state}
+                  .stateObj=${state}
+                  .hass=${this._hass}
                   ?data-domain=${domain}
                   data-state=${ifDefined(state?.state)}
                   style=${styleMap(haIconStyle)}