Fixed regression training bug