Fixed bug preventing train switching in TrainControllerUI