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