Fix puppet script