bug fix for training scale issue