Fix zookeeper puppet scripts