fixed training label issue