fixes to training