1 files changed, 5 insertions, 9 deletions
diff --git a/manifests/lastlog/disable.pp b/manifests/lastlog/disable.pp
index 519195f..e143477 100644
--- a/manifests/lastlog/disable.pp
+++ b/manifests/lastlog/disable.pp
@@ -1,13 +1,9 @@
-class loginrecords::lastlog::disable(
-  $pam_login_file = '/etc/pam.d/login'
-){
-  replace { 'loginrecords-lastlog-disable':
-    file    => $pam_login_file,
+class loginrecords::lastlog::disable inherits loginrecords::lastlog::enable {
+  Replace['loginrecords-lastlog']{
     pattern   => '^session[[:space:]]+optional[[:space:]]+pam_lastlog.so$',
     replacement => '#session  optional   pam_lastlog.so',
   }
-  file{'/var/log/lastlog':
-    content => '',
-    owner => root, group => utmp, mode => 0664;
-  } 
+  File['/var/log/lastlog']{
+    ensure => absent,
+  }
 }