Fix comprehension