Fix getting the current train