Remove flatten