flatmap fix