diff --git a/homeassistant/components/rainmachine/config_flow.py b/homeassistant/components/rainmachine/config_flow.py
index 6b11e82d023..39884681967 100644
--- a/homeassistant/components/rainmachine/config_flow.py
+++ b/homeassistant/components/rainmachine/config_flow.py
@@ -70,7 +70,7 @@ class RainMachineFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
         self, discovery_info: zeroconf.ZeroconfServiceInfo
     ) -> FlowResult:
         """Handle discovery via zeroconf."""
-        ip_address = discovery_info[zeroconf.ATTR_HOST]
+        ip_address = discovery_info.host
 
         self._async_abort_entries_match({CONF_IP_ADDRESS: ip_address})
         # Handle IP change