Doesn't always show multiple departures

[djmc]

There seems to be an issue with loading the right configuration for how many departures to display.

I have an entity for my local bus stop and I want to show the next 2 departures. It's configured using the UI (ie not YAML)

It works when I first set it up, but when reloading my HA, it no longer shows the additional departures (only the next one).

To fix it I need to Reload the integration until it actually shows the departure.

I tried figuring out the fix myself but I couldn't quite work out what's causing it. I suspect something to do with loading configuration due to how reloading fixes it, but looking at the code it makes sense to me but I don't know how the HA API works.