Fix for the live trains query