MO flatten bug fix