Fix flatten bug