Add missing brace to comprehension