2 files changed, 11 insertions, 6 deletions
diff --git a/manifests/utmp/disable.pp b/manifests/utmp/disable.pp
index 8d9145a..5b4ce07 100644
--- a/manifests/utmp/disable.pp
+++ b/manifests/utmp/disable.pp
@@ -1,3 +1,5 @@
-class loginrecords::utmp::disable {
-    file { "$utmp_file": ensure => 'absent' }
+class loginrecords::utmp::disable inherits loginrecords::utmp::enable {
+  File[$utmp_file]{
+    ensure => 'absent'
+  }
 }
diff --git a/manifests/utmp/enable.pp b/manifests/utmp/enable.pp
index 81e58ec..ee47b94 100644
--- a/manifests/utmp/enable.pp
+++ b/manifests/utmp/enable.pp
@@ -1,5 +1,8 @@
-class loginrecords::utmp::enable {
-    file { "$utmp_file":
-        ensure => 'present', mode => 660, owner => 'root', group => 'utmp',
-    }
+class loginrecords::utmp::enable(
+  $utmp_file = '/var/run/utmp'
+){
+  file {$utmp_file:
+    ensure => 'present',
+    owner => 'root', group => 'utmp', mode => 660;
+  }
 }