Fix DistributionFactory usage