It seems we don't need the flat_map