diff --git a/homeassistant/components/neato/entity.py b/homeassistant/components/neato/entity.py
index 72eac57127b4d3..43072f1969391f 100644
--- a/homeassistant/components/neato/entity.py
+++ b/homeassistant/components/neato/entity.py
@@ -3,7 +3,8 @@
 
 from pybotvac import Robot
 
-from homeassistant.helpers.entity import DeviceInfo, Entity
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.entity import Entity
 
 from .const import NEATO_DOMAIN