From 78940c45af35a4d2f3e3aaeb00d45ceb7ac39627 Mon Sep 17 00:00:00 2001 From: James Connor Date: Tue, 23 Aug 2022 11:31:29 +0100 Subject: [PATCH] Fix for --- custom_components/realtime_trains_api/sensor.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/custom_components/realtime_trains_api/sensor.py b/custom_components/realtime_trains_api/sensor.py index a603dcb..b38466e 100644 --- a/custom_components/realtime_trains_api/sensor.py +++ b/custom_components/realtime_trains_api/sensor.py @@ -194,8 +194,10 @@ async def _async_update(self): "platform": departure["locationDetail"].get("platform", None), "operator_name": departure["atocName"], } - if departureCount <= self._journey_data_for_next_X_trains: - await self._add_journey_data(train, scheduledTs, estimatedTs) + if departureCount > self._journey_data_for_next_X_trains: + break; + + await self._add_journey_data(train, scheduledTs, estimatedTs) self._next_trains.append(train) if nextDepartureEstimatedTs is None: