diff --git a/source/_cookbook/restart_ha_if_wemo_switch_is_not_detected.markdown b/source/_cookbook/restart_ha_if_wemo_switch_is_not_detected.markdown
index 7a1beb312eb..3caf90f7caf 100644
--- a/source/_cookbook/restart_ha_if_wemo_switch_is_not_detected.markdown
+++ b/source/_cookbook/restart_ha_if_wemo_switch_is_not_detected.markdown
@@ -29,9 +29,11 @@ notify:
   - platform: pushbullet
     api_key: ***
     name: pushbullet
-  
+
+wemo:
+  discovery: true
+
 switch:
-  - platform: wemo
   - platform: mqtt
     state_topic: "home/killhass"
     command_topic: "home/killhass"