Little baby hotfix

Just a small update to fix the training targets not working. :)