PUPPET-142 Update tests