fix training.js