Updating distributor code to handle puppet distributors