diff --git a/drivers/SmartThings/zigbee-switch/fingerprints.yml b/drivers/SmartThings/zigbee-switch/fingerprints.yml
index d87b025ce0..8146b123a1 100644
--- a/drivers/SmartThings/zigbee-switch/fingerprints.yml
+++ b/drivers/SmartThings/zigbee-switch/fingerprints.yml
@@ -2150,22 +2150,22 @@ zigbeeManufacturer:
     deviceLabel: 智能墙面五孔插座
     manufacturer: WALL HERO
     model: ACL-401ON
-    deviceProfileName: basic-switch    
+    deviceProfileName: wallhero    
   - id: "WALL HERO/ACL-401S1I"
     deviceLabel: 一位智能开关面板
     manufacturer: WALL HERO
     model: ACL-401S1I
-    deviceProfileName: basic-switch    
+    deviceProfileName: wallhero    
   - id: "WALL HERO/ACL-401S3I"
     deviceLabel: 三位智能开关面板 1
     manufacturer: WALL HERO
     model: ACL-401S3I
-    deviceProfileName: basic-switch    
+    deviceProfileName: wallhero    
   - id: "WALL HERO/ACL-401S2I"
     deviceLabel: 二位智能开关面板 1
     manufacturer: WALL HERO
     model: ACL-401S2I
-    deviceProfileName: basic-switch
+    deviceProfileName: wallhero
   # End Wall Hero
   - id: "Insta GmbH/Switching Actuator"
     deviceLabel: NEXENTRO Switching Actuator
diff --git a/drivers/SmartThings/zigbee-switch/src/wallhero/init.lua b/drivers/SmartThings/zigbee-switch/src/wallhero/init.lua
index fb7f1a2e36..c96e89a16d 100644
--- a/drivers/SmartThings/zigbee-switch/src/wallhero/init.lua
+++ b/drivers/SmartThings/zigbee-switch/src/wallhero/init.lua
@@ -28,7 +28,9 @@ local FINGERPRINTS = {
   { mfr = "WALL HERO", model = "ACL-401S4I", switches = 4, buttons = 0 },
   { mfr = "WALL HERO", model = "ACL-401S8I", switches = 4, buttons = 4 },
   { mfr = "WALL HERO", model = "ACL-401S3I", switches = 3, buttons = 0 },
-  { mfr = "WALL HERO", model = "ACL-401S2I", switches = 2, buttons = 0 }
+  { mfr = "WALL HERO", model = "ACL-401S2I", switches = 2, buttons = 0 },
+  { mfr = "WALL HERO", model = "ACL-401S1I", switches = 1, buttons = 0 },
+  { mfr = "WALL HERO", model = "ACL-401ON", switches = 1, buttons = 0 }
 }
 
 local function can_handle_wallhero_switch(opts, driver, device, ...)