One more list comprehension simplification