Reverted Puppet Stage code to working syntax