diff --git a/homeassistant/components/flo/device.py b/homeassistant/components/flo/device.py index 3b7469686b45..0c805b932cba 100644 --- a/homeassistant/components/flo/device.py +++ b/homeassistant/components/flo/device.py @@ -46,7 +46,7 @@ class FloDeviceDataUpdateCoordinator(DataUpdateCoordinator): # pylint: disable= await self._update_device() await self._update_consumption_data() self._failure_count = 0 - except RequestError as error: + except (RequestError, TimeoutError) as error: self._failure_count += 1 if self._failure_count > 3: raise UpdateFailed(error) from error diff --git a/tests/components/flo/test_device.py b/tests/components/flo/test_device.py index 6a633c774ede..eb7785dbd616 100644 --- a/tests/components/flo/test_device.py +++ b/tests/components/flo/test_device.py @@ -90,7 +90,7 @@ async def test_device( # test error sending device ping with patch( - "homeassistant.components.flo.device.FloDeviceDataUpdateCoordinator.send_presence_ping", + "aioflo.presence.Presence.ping", side_effect=RequestError, ), pytest.raises(UpdateFailed): # simulate 4 updates failing