1 files changed, 9 insertions, 9 deletions
diff --git a/manifests/lastlog/enable.pp b/manifests/lastlog/enable.pp
index b8dec35..cb6e8a6 100644
--- a/manifests/lastlog/enable.pp
+++ b/manifests/lastlog/enable.pp
@@ -2,15 +2,15 @@
 class loginrecords::lastlog::enable(
   $pam_login_file = '/etc/pam.d/login'
 ){
-  replace{'loginrecords-lastlog':
-    file        => $pam_login_file,
-    pattern     => '^#session[[:space:]]+optional[[:space:]]+pam_lastlog.so$',
-    replacement => 'session  optional   pam_lastlog.so',
+  file_line{ 'loginrecords-lastlog':
+    path  => $pam_login_file,
+    match => '^.*session[[:space:]]+optional[[:space:]]+pam_lastlog.so$',
+    line  => 'session  optional   pam_lastlog.so',
   }
-  file{'/var/log/lastlog':
-    ensure  => present,
-    owner   => 'root',
-    group   => 'utmp',
-    mode    => '0664';
+  file{ '/var/log/lastlog':
+    ensure => present,
+    owner  => root,
+    group  => utmp,
+    mode   => '0664';
   }
 }