diff --git a/sonoff_zigbee_button.yaml b/sonoff_zigbee_button.yaml
index 96bb177..40a8300 100644
--- a/sonoff_zigbee_button.yaml
+++ b/sonoff_zigbee_button.yaml
@@ -13,7 +13,7 @@ blueprint:
           - manufacturer: eWeLink
             model: WB01
           - manufacturer: eWeLink
-            model: SNZB-01
+            model: SNZB-01P
           multiple: false
     mode:
       name: Automation mode