Fixed broken list comprehension