fix training
test