Modified fit section according to --fit-training.