Fixed a major bug in list comprehension