Use the correct fitness function