diff --git a/custom_components/battery_notes/sensor.py b/custom_components/battery_notes/sensor.py
index 26c26b779..a97287275 100644
--- a/custom_components/battery_notes/sensor.py
+++ b/custom_components/battery_notes/sensor.py
@@ -132,6 +132,8 @@ async def async_setup_entry(
     entity_registry = er.async_get(hass)
     device_registry = dr.async_get(hass)
 
+    device_id = config_entry.data.get(CONF_DEVICE_ID, None)
+
     async def async_registry_updated(event: Event) -> None:
         """Handle entity registry update."""
         data = event.data