fix train-app layout