Bugfix for switch

iMicknl · Jul 21, 2020 · 01c1df1 · 01c1df1
1 parent a13348f
commit 01c1df1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/custom_components/tahoma/switch.py b/custom_components/tahoma/switch.py
@@ -26,9 +26,7 @@ async def async_setup_entry(hass, entry, async_add_entities):
     controller = data.get("controller")
 
     entities = [
-        TahomaSwitch(device, controller)
-        for device in data.get("devices")
-        if TAHOMA_TYPES[device.ui_class] == "switch"
+        TahomaSwitch(device, controller) for device in data.get("entities").get("switch")
     ]
 
     async_add_entities(entities)