Tag: fix training errors