fix nested list comprehension