fix train display problem