diff --git a/manifests/lib/docs.pp b/manifests/lib/docs.pp
index 78ea448..4b92b77 100644
--- a/manifests/lib/docs.pp
+++ b/manifests/lib/docs.pp
@@ -1,14 +1,14 @@
 # This class installs the postgresql-docs See README.md for more
 # details.
 class postgresql::lib::docs (
-  String $package_name   = $postgresql::params::docs_package_name,
-  $package_ensure = 'present',
+  String $package_name                                             = $postgresql::params::docs_package_name,
+  Enum['present', 'absent', 'latest', 'installed'] $package_ensure = 'present',
 ) inherits postgresql::params {
 
   package { 'postgresql-docs':
     ensure => $package_ensure,
     name   => $package_name,
     tag    => 'postgresql',
   }
 
 }