From ab599cd59dea61599f5b16588852281671468318 Mon Sep 17 00:00:00 2001 From: andvikt Date: Thu, 28 Oct 2021 20:14:30 +0300 Subject: [PATCH] . --- custom_components/mega/entities.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/custom_components/mega/entities.py b/custom_components/mega/entities.py index 2fae4c8..2375c3e 100644 --- a/custom_components/mega/entities.py +++ b/custom_components/mega/entities.py @@ -129,7 +129,7 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity): if self.hass is None: return {} if self._customize is None: - c_entity_id = self.hass.data.get(DOMAIN, {}).get(self.entity_id, {}) + c_entity_id = self.hass.data.get(DOMAIN, {}).get('entities', {}).get(self.entity_id, {}) c = self.hass.data.get(DOMAIN, {}).get(CONF_CUSTOM) or {} c = c.get(self._mega_id) or {} c = c.get(self.port) or {}