FlatnessLoss changed a bit