diff --git a/custom_components/mega/sensor.py b/custom_components/mega/sensor.py index b298a0b..dd52844 100644 --- a/custom_components/mega/sensor.py +++ b/custom_components/mega/sensor.py @@ -171,7 +171,12 @@ class MegaI2C(FilterBadValues): @property def customize(self): - return super().customize.get(self.id_suffix, {}) or {} + ret = super().customize + _old = ret.get(self.id_suffix) + if _old is not None: + ret = ret.copy() + ret.update(_old) + return ret @property def extra_state_attributes(self):