Keep list comprehension on one line