Update switch.py

veista · Mar 28, 2022 · 9f2dc53 · 9f2dc53
1 parent 202ca69
commit 9f2dc53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/custom_components/smartthings/switch.py b/custom_components/smartthings/switch.py
@@ -66,7 +66,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
     for device in broker.devices.values():
         for capability in broker.get_assigned(device.device_id, "switch"):
             maps = CAPABILITY_TO_SWITCH[capability]
-            if capability == "custom.autoCleaningMode":
+            if capability in ("custom.autoCleaningMode", "custom.spiMode"):
                 switches.extend(
                     [
                         SmartThingsCustomSwitch(