Fixing flattening part 1.