fix list comprehension