fix train data and train labels