diff --git a/homeassistant/components/device_automation/__init__.py b/homeassistant/components/device_automation/__init__.py
index 68d05c19f67..979c82acfe2 100644
--- a/homeassistant/components/device_automation/__init__.py
+++ b/homeassistant/components/device_automation/__init__.py
@@ -247,9 +247,10 @@ async def async_get_device_automations(
     match_device_ids = set(device_ids or device_registry.devices)
     combined_results: dict[str, list[dict[str, Any]]] = {}
 
-    for entry in entity_registry.entities.values():
-        if not entry.disabled_by and entry.device_id in match_device_ids:
-            device_entities_domains.setdefault(entry.device_id, set()).add(entry.domain)
+    for device_id in match_device_ids:
+        for entry in entity_registry.entities.get_entries_for_device_id(device_id):
+            if not entry.disabled_by:
+                device_entities_domains.setdefault(device_id, set()).add(entry.domain)
 
     for device_id in match_device_ids:
         combined_results[device_id] = []