Minor: using list comprehension