Bug fix in find trains