2 files changed, 11 insertions, 6 deletions
diff --git a/manifests/btmp/disable.pp b/manifests/btmp/disable.pp
index 809c669..7a27126 100644
--- a/manifests/btmp/disable.pp
+++ b/manifests/btmp/disable.pp
@@ -1,4 +1,6 @@
-class loginrecords::btmp::disable {
-    file { "$btmp_file": ensure => 'absent' }
+class loginrecords::btmp::disable inherits loginrecords::btmp::enable {
+  File[$btmp_file]{
+    ensure => 'absent'
+  }
 }
 
diff --git a/manifests/btmp/enable.pp b/manifests/btmp/enable.pp
index ea19e16..c173362 100644
--- a/manifests/btmp/enable.pp
+++ b/manifests/btmp/enable.pp
@@ -1,5 +1,8 @@
-class loginrecords::btmp::enable {
-    file { "$btmp_file":
-        ensure => 'present', mode => 660, owner => 'root', group => 'utmp',
-    }
+class loginrecords::btmp::enable(
+  $btmp_file = '/var/log/btmp'
+){
+  file{$btmp_file:
+    ensure => 'present',
+    owner => 'root', group => 'utmp', mode => 660;
+  }
 }