From 9f2dc539989259a5c49ce617fc13bddf8c4f448a Mon Sep 17 00:00:00 2001
From: veista <91916713+veista@users.noreply.github.com>
Date: Tue, 29 Mar 2022 01:29:46 +0300
Subject: [PATCH] Update switch.py

---
 custom_components/smartthings/switch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_components/smartthings/switch.py b/custom_components/smartthings/switch.py
index ff3a276..3c43697 100644
--- 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(