Quick fix for breakage with updated distribute

Going to just self-merge this, since it seems to make tests happy again after a change to distribute