Fixes to puppet script.