diff --git a/custom_components/hhs_vertretungsplan/binary_sensor.py b/custom_components/hhs_vertretungsplan/binary_sensor.py index a362b45..b0522ac 100644 --- a/custom_components/hhs_vertretungsplan/binary_sensor.py +++ b/custom_components/hhs_vertretungsplan/binary_sensor.py @@ -87,8 +87,9 @@ def extra_state_attributes(self) -> Dict[str, Any]: ] vertretungen = self.coordinator.data[ATTR_VERTRETUNG] _LOGGER.debug(f"before adding: len(vertretung_state) = {len(vertretung_state)}") - vertretung_state = [vertretungen[x] for x in selection if x in vertretungen.keys()] - + for select in selection: + if select in vertretungen.keys(): + vertretung_state.extend(vertretungen[select].copy()) _LOGGER.debug(f"after adding: len(vertretung_state) = {len(vertretung_state)}") return {