Can you change this `collect` to `map` for consistency?