Bug in flattening correction is fixed