nit: I don't think the list comprehension is necessary here (it'll work as a generator)