I'm going to propose rewriting the get_units to return a generator as part of this distributor re-write.