Fixed training issue.