From 95d6fd0b9bad3667687a31d224c40e41a9ee1607 Mon Sep 17 00:00:00 2001
From: cyr-ius <cyr-ius@ipocus.net>
Date: Tue, 10 Dec 2024 08:43:45 +0100
Subject: [PATCH] Fix coodinator data entry

---
 custom_components/livebox/diagnostics.py | 4 ++--
 custom_components/livebox/manifest.json  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/custom_components/livebox/diagnostics.py b/custom_components/livebox/diagnostics.py
index f917fa8..edbf7e2 100644
--- a/custom_components/livebox/diagnostics.py
+++ b/custom_components/livebox/diagnostics.py
@@ -1,4 +1,5 @@
 """Diagnostics support for Livebox."""
+
 from __future__ import annotations
 
 import logging
@@ -9,7 +10,6 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 
-from .const import DOMAIN
 
 TO_REDACT = {
     "address",
@@ -112,7 +112,7 @@ async def async_get_config_entry_diagnostics(
     hass: HomeAssistant, entry: ConfigEntry
 ) -> dict[str, Any]:
     """Return diagnostics for a config entry."""
-    coordinator = hass.data[DOMAIN][entry.entry_id]
+    coordinator = entry.runtime_data
     api_methods = [
         coordinator.api.async_get_permissions,
         coordinator.api.deviceinfo.async_get_deviceinfo,
diff --git a/custom_components/livebox/manifest.json b/custom_components/livebox/manifest.json
index adf28d2..4bc1206 100644
--- a/custom_components/livebox/manifest.json
+++ b/custom_components/livebox/manifest.json
@@ -15,5 +15,5 @@
       "friendlyName": "Orange Livebox"
     }
   ],
-  "version": "2.3.8"
+  "version": "2.3.9"
 }