Puppet script fix