code review

same here, use `map`