diff --git a/homeassistant/components/lamarzocco/update.py b/homeassistant/components/lamarzocco/update.py index 33e64623256..3dbce542ff3 100644 --- a/homeassistant/components/lamarzocco/update.py +++ b/homeassistant/components/lamarzocco/update.py @@ -125,7 +125,7 @@ class LaMarzoccoUpdateEntity(LaMarzoccoEntity, UpdateEntity): await self.coordinator.device.update_firmware() while ( update_progress := await self.coordinator.device.get_firmware() - ).command_status is UpdateStatus.IN_PROGRESS: + ).command_status is not UpdateStatus.UPDATED: if counter >= MAX_UPDATE_WAIT: _raise_timeout_error() self._attr_update_percentage = update_progress.progress_percentage diff --git a/tests/components/lamarzocco/test_update.py b/tests/components/lamarzocco/test_update.py index 99f85c21381..42458902411 100644 --- a/tests/components/lamarzocco/test_update.py +++ b/tests/components/lamarzocco/test_update.py @@ -62,7 +62,7 @@ async def test_update_process( ), UpdateDetails( status=UpdateStatus.UPDATED, - command_status=None, + command_status=UpdateStatus.UPDATED, progress_info=None, progress_percentage=None, ),