fix missing setDistribution