fix to the fitness function