diff --git a/homeassistant/components/lamarzocco/binary_sensor.py b/homeassistant/components/lamarzocco/binary_sensor.py index c108bdb02d8..aacfca929ad 100644 --- a/homeassistant/components/lamarzocco/binary_sensor.py +++ b/homeassistant/components/lamarzocco/binary_sensor.py @@ -61,7 +61,10 @@ ENTITIES: tuple[LaMarzoccoBinarySensorEntityDescription, ...] = ( device_class=BinarySensorDeviceClass.RUNNING, is_on_fn=( lambda machine: cast( - BackFlush, machine.dashboard.config[WidgetType.CM_BACK_FLUSH] + BackFlush, + machine.dashboard.config.get( + WidgetType.CM_BACK_FLUSH, BackFlush(status=BackFlushStatus.OFF) + ), ).status is BackFlushStatus.REQUESTED ), diff --git a/homeassistant/components/lamarzocco/sensor.py b/homeassistant/components/lamarzocco/sensor.py index aecb2ff7f04..afe34005108 100644 --- a/homeassistant/components/lamarzocco/sensor.py +++ b/homeassistant/components/lamarzocco/sensor.py @@ -5,7 +5,7 @@ from dataclasses import dataclass from datetime import datetime from typing import cast -from pylamarzocco.const import ModelName, WidgetType +from pylamarzocco.const import BackFlushStatus, ModelName, WidgetType from pylamarzocco.models import ( BackFlush, BaseWidgetOutput, @@ -106,7 +106,10 @@ ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...] = ( device_class=SensorDeviceClass.TIMESTAMP, value_fn=( lambda config: cast( - BackFlush, config[WidgetType.CM_BACK_FLUSH] + BackFlush, + config.get( + WidgetType.CM_BACK_FLUSH, BackFlush(status=BackFlushStatus.OFF) + ), ).last_cleaning_start_time ), entity_category=EntityCategory.DIAGNOSTIC,