diff --git a/lovelace/overview/energy.yaml b/lovelace/overview/energy.yaml
index 0d050bf..f034da7 100755
--- a/lovelace/overview/energy.yaml
+++ b/lovelace/overview/energy.yaml
@@ -57,7 +57,7 @@
         ulm_card_generic_icon: "mdi:home-lightning-bolt"
     - type: 'custom:button-card'
       template: card_generic
-      entity: switch.kai_wifi_power
+      entity: switch.kai_wifi
       variables:
         ulm_card_generic_name: WiFi
         ulm_card_generic_icon: "mdi:home-lightning-bolt"