diff --git a/config/lovelace/button_card_templates/half_button.yaml b/config/lovelace/button_card_templates/half_button.yaml
index 02220f5..3109586 100644
--- a/config/lovelace/button_card_templates/half_button.yaml
+++ b/config/lovelace/button_card_templates/half_button.yaml
@@ -1,4 +1,5 @@
 half:
+  template: with_state
   size: 30px
   show_icon: false
   hold_action:
@@ -7,18 +8,3 @@ half:
     card:
       - height: 45px
       - font-size: 14px
-  state:
-    - value: 'on'
-      styles:
-        card:
-          - background-color: var(--primary-color)
-          - color: black
-        icon:
-          - color: black
-    - value: 'home'
-      styles:
-        card:
-          - background-color: var(--primary-color)
-          - color: black
-        icon:
-          - color: black