Update module to be compatible with Puppet 3