diff --git a/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml b/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml
index 75bfc76..487d003 100644
--- a/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml
+++ b/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml
@@ -208,5 +208,6 @@ time:
 web_server:
 
 wifi:
+  id: wifi_component
   ap:
 ...
diff --git a/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml b/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml
index 56d72da..afed401 100644
--- a/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml
+++ b/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml
@@ -192,7 +192,7 @@ script:
 
           // Toggle relay if corresponding button action is enabled
           const bool api_disconnected = !global_api_server->is_connected() or
-                                        !global_wifi_component->is_connected();
+                                        !wifi_component->is_connected();
           std::string select_state;
           bool toggle_relay = false;
           if (api_disconnected) {