diff --git a/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml b/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml
index 5735035..56d72da 100644
--- a/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml
+++ b/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_relays.yaml
@@ -191,7 +191,8 @@ script:
           }
 
           // Toggle relay if corresponding button action is enabled
-          const bool api_disconnected = not global_api_server->is_connected();
+          const bool api_disconnected = !global_api_server->is_connected() or
+                                        !global_wifi_component->is_connected();
           std::string select_state;
           bool toggle_relay = false;
           if (api_disconnected) {