Fixed finish training bug