Add lower switch to the flattening pass