Restore list comprehension from generator syntax