Fix for

megakid · Aug 23, 2022 · 78940c4 · 78940c4
1 parent 2fc3ed6
commit 78940c4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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: