fix train set