fixing no-training bug