diff --git a/homeassistant/components/bang_olufsen/util.py b/homeassistant/components/bang_olufsen/util.py
index d869c63efa2a..4203417424d4 100644
--- a/homeassistant/components/bang_olufsen/util.py
+++ b/homeassistant/components/bang_olufsen/util.py
@@ -21,10 +21,12 @@ from .const import (
 )
 
 
-def get_device(hass: HomeAssistant, unique_id: str) -> DeviceEntry:
+def get_device(hass: HomeAssistant, unique_id: str, entry_id: str) -> DeviceEntry:
     """Get the device."""
     device_registry = dr.async_get(hass)
-    device = device_registry.async_get_device({(DOMAIN, unique_id)})
+    device = device_registry.async_get_device_by_identifier(
+        (DOMAIN, unique_id), entry_id
+    )
     assert device
 
     return device
diff --git a/homeassistant/components/bang_olufsen/websocket.py b/homeassistant/components/bang_olufsen/websocket.py
index 93e352b47441..0ed29ed916f6 100644
--- a/homeassistant/components/bang_olufsen/websocket.py
+++ b/homeassistant/components/bang_olufsen/websocket.py
@@ -50,7 +50,7 @@ class BeoWebsocket(BeoBase):
         BeoBase.__init__(self, entry, client)
 
         self.hass = hass
-        self._device = get_device(hass, self._unique_id)
+        self._device = get_device(hass, self._unique_id, self.entry.entry_id)
 
         # WebSocket callbacks
         self._client.get_notification_notifications(self.on_notification_notification)