2 files changed, 11 insertions, 6 deletions
diff --git a/manifests/wtmp/disable.pp b/manifests/wtmp/disable.pp
index c7c2b6d..c974cd9 100644
--- a/manifests/wtmp/disable.pp
+++ b/manifests/wtmp/disable.pp
@@ -1,3 +1,5 @@
-class loginrecords::wtmp::disable {
-    file { "$wtmp_file": ensure => 'absent' }
+class loginrecords::wtmp::disable inherits loginrecords::wtmp::enable {
+  File[$wtmp_filer]{
+    ensure => 'absent'
+  }
 }
diff --git a/manifests/wtmp/enable.pp b/manifests/wtmp/enable.pp
index 4982839..f3b5ee9 100644
--- a/manifests/wtmp/enable.pp
+++ b/manifests/wtmp/enable.pp
@@ -1,5 +1,8 @@
-class loginrecords::wtmp::enable {
-    file { "$wtmp_file":
-        ensure => 'present', mode => 664, owner => 'root', group => 'utmp',
-    }
+class loginrecords::wtmp::enable(
+  $wtmp_file = '/var/log/wtmp'
+){
+  file{$wtmp_file:
+    ensure => 'present',
+    owner => 'root', group => 'utmp', mode => 664;
+  }
 }