adding change to fix training