Update TrainControllerModule.java