Fix invalid dict comprehension