I think you should just use collect() here.