Fix smaller list comprehension