Fix unit tests in recent Puppet 3 versions