Update comprehension as there is a default arg