diff --git a/homeassistant/components/binary_sensor/wemo.py b/homeassistant/components/binary_sensor/wemo.py index f3cd91baa04..0e3259a3a96 100644 --- a/homeassistant/components/binary_sensor/wemo.py +++ b/homeassistant/components/binary_sensor/wemo.py @@ -45,6 +45,9 @@ class WemoBinarySensor(BinarySensorDevice): _LOGGER.info( 'Subscription update for %s', _device) + if not hasattr(self, 'hass'): + self.update() + return self.update_ha_state(True) @property diff --git a/homeassistant/components/switch/wemo.py b/homeassistant/components/switch/wemo.py index 90d0f46ba68..63b2665449e 100644 --- a/homeassistant/components/switch/wemo.py +++ b/homeassistant/components/switch/wemo.py @@ -63,6 +63,9 @@ class WemoSwitch(SwitchDevice): _LOGGER.info( 'Subscription update for %s', _device) + if not hasattr(self, 'hass'): + self.update() + return self.update_ha_state(True) @property