Added canUseFlatMap test