diff --git a/homeassistant/components/wled/config_flow.py b/homeassistant/components/wled/config_flow.py index b2112e8962b..485afef4f6c 100644 --- a/homeassistant/components/wled/config_flow.py +++ b/homeassistant/components/wled/config_flow.py @@ -44,14 +44,14 @@ class WLEDFlowHandler(ConfigFlow, domain=DOMAIN): """Handle zeroconf discovery.""" # Hostname is format: wled-livingroom.local. - host = discovery_info[zeroconf.ATTR_HOSTNAME].rstrip(".") + host = discovery_info.hostname.rstrip(".") name, _ = host.rsplit(".") self.context.update( { - CONF_HOST: discovery_info[zeroconf.ATTR_HOST], + CONF_HOST: discovery_info.host, CONF_NAME: name, - CONF_MAC: discovery_info[zeroconf.ATTR_PROPERTIES].get(CONF_MAC), + CONF_MAC: discovery_info.properties.get(CONF_MAC), "title_placeholders": {"name": name}, } )