diff --git a/tasks/init.rb b/tasks/init.rb
index 4305312..86367d9 100755
--- a/tasks/init.rb
+++ b/tasks/init.rb
@@ -1,26 +1,33 @@
 #!/opt/puppetlabs/puppet/bin/ruby
 # frozen_string_literal: true
 
 require 'json'
 require 'open3'
 require 'puppet'
 
 def apt_get(action)
   cmd = ['apt-get', action]
   cmd << '-y' if ['upgrade', 'dist-upgrade', 'autoremove'].include?(action)
+  if ['upgrade', 'dist-upgrade'].include?(action)
+    ENV['DEBIAN_FRONTEND'] = 'noninteractive'
+    cmd << '-o'
+    cmd << 'Dpkg::Options="--force-confdef"'
+    cmd << '-o'
+    cmd << 'Dpkg::Options="--force-confold"'
+  end
   stdout, stderr, status = Open3.capture3(*cmd)
   raise Puppet::Error, stderr if status != 0
   { status: stdout.strip }
 end
 
 params = JSON.parse(STDIN.read)
 action = params['action']
 
 begin
   result = apt_get(action)
   puts result.to_json
   exit 0
 rescue Puppet::Error => e
   puts({ status: 'failure', error: e.message }.to_json)
   exit 1
 end