fix flatMap test