Fixed flatten bug.