Rewritten to use list comprehension instead of map

One of the places where a map() was used was rewritten to use a list
comprehension