Adjust docs and return type for when_all_n algning things with latest changes on master