list comprehension mistake