Fix old import

bang-olufsen · Feb 24, 2023 · 1cb31cd · 1cb31cd
1 parent ef06f8d
commit 1cb31cd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/custom_components/bangolufsen/button.py b/custom_components/bangolufsen/button.py
@@ -18,6 +18,7 @@
     HASS_FAVOURITES,
     BangOlufsenVariables,
     get_device,
+    generate_favourite_attributes,
 )
 from .coordinator import BangOlufsenCoordinator
 
@@ -111,7 +112,7 @@ async def async_added_to_hass(self) -> None:
             self.coordinator.async_add_listener(self._update_favourite)
         )
 
-        self._attr_extra_state_attributes = self.generate_favourite_attributes(
+        self._attr_extra_state_attributes = generate_favourite_attributes(
             self._favourite
         )
 
@@ -126,7 +127,7 @@ def _update_favourite(self) -> None:
         self._favourite = self.coordinator.data["favourites"][str(self._favourite_id)]
 
         if old_favourite != self._favourite:
-            self._attr_extra_state_attributes = self.generate_favourite_attributes(
+            self._attr_extra_state_attributes = generate_favourite_attributes(
                 self._favourite
             )