Why flatten here?