fixes to make collect/mean_reduce work