diff --git a/homeassistant/components/unifi/switch.py b/homeassistant/components/unifi/switch.py
index 5b64f573ccd..b1f62131eb4 100644
--- a/homeassistant/components/unifi/switch.py
+++ b/homeassistant/components/unifi/switch.py
@@ -42,12 +42,15 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
 
             _, mac = entity.unique_id.split("-", 1)
 
-            if mac in controller.api.clients or mac not in controller.api.clients_all:
+            if mac in controller.api.clients:
+                switches_off.append(entity.unique_id)
                 continue
 
-            client = controller.api.clients_all[mac]
-            controller.api.clients.process_raw([client.raw])
-            switches_off.append(entity.unique_id)
+            if mac in controller.api.clients_all:
+                client = controller.api.clients_all[mac]
+                controller.api.clients.process_raw([client.raw])
+                switches_off.append(entity.unique_id)
+                continue
 
     @callback
     def update_controller():