WIP: Puppet agent tests -- fix