diff --git a/custom_components/battery_notes/device.py b/custom_components/battery_notes/device.py
index 69787591c..da05b623e 100644
--- a/custom_components/battery_notes/device.py
+++ b/custom_components/battery_notes/device.py
@@ -85,10 +85,7 @@ async def async_setup(self) -> bool:
         for entity in entity_registry.entities.values():
             if not entity.device_id or entity.device_id != device_id:
                 continue
-            if not entity.domain or entity.domain not in {
-                BINARY_SENSOR_DOMAIN,
-                SENSOR_DOMAIN,
-            }:
+            if not entity.domain or entity.domain != SENSOR_DOMAIN:
                 continue
             if not entity.platform or entity.platform == DOMAIN:
                 continue