Fixing syntax for comprehension