Fix list comprehension