Fix comprehension mistake