diff --git a/manifests/init.pp b/manifests/init.pp
index ab4d2b4..2d30dcd 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -1,84 +1,84 @@
 class unattended_upgrades (
   Unattended_upgrades::Age                  $age                    = {},
   Unattended_upgrades::Auto                 $auto                   = {},
   Unattended_upgrades::Backup               $backup                 = {},
   Array[String[1]]                          $blacklist              = [],
   Array[String[1]]                          $whitelist              = [],
   Optional[Integer[0]]                      $dl_limit               = undef,
   Integer[0, 1]                             $enable                 = 1,
   Boolean                                   $install_on_shutdown    = false,
   Unattended_upgrades::Mail                 $mail                   = {},
   Boolean                                   $minimal_steps          = true,
   Array[Unattended_upgrades::Origin]        $origins                = $unattended_upgrades::params::origins,
   String[1]                                 $package_ensure         = installed,
   Array[String[1]]                          $extra_origins          = [],
   Optional[Integer[0]]                      $random_sleep           = undef,
   Optional[String]                          $sender                 = undef,
   Integer[0]                                $size                   = 0,
-  Integer[0]                                $update                 = 1,
-  Integer[0]                                $upgrade                = 1,
+  Variant[Integer[0], Enum['always']]       $update                 = 1,
+  Variant[Integer[0], Enum['always']]       $upgrade                = 1,
   Unattended_upgrades::Upgradeable_packages $upgradeable_packages   = {},
   Integer[0]                                $verbose                = 0,
   Boolean                                   $notify_update          = false,
   Unattended_upgrades::Options              $options                = {},
   Array[String[1]]                          $days                   = [],
   Optional[Boolean]                         $remove_unused_kernel   = undef,
   Optional[Boolean]                         $remove_new_unused_deps = undef,
   Optional[Boolean]                         $syslog_enable          = undef,
   Optional[String]                          $syslog_facility        = undef,
 ) inherits unattended_upgrades::params {
   # apt::conf settings require the apt class to work
   include apt
 
   $_age = merge($unattended_upgrades::default_age, $age)
   assert_type(Unattended_upgrades::Age, $_age)
 
   $_auto = merge($unattended_upgrades::default_auto, $auto)
   assert_type(Unattended_upgrades::Auto, $_auto)
 
   $_backup = merge($unattended_upgrades::default_backup, $backup)
   assert_type(Unattended_upgrades::Backup, $_backup)
 
   $_mail = merge($unattended_upgrades::default_mail, $mail)
   assert_type(Unattended_upgrades::Mail, $_mail)
 
   $_upgradeable_packages = merge($unattended_upgrades::default_upgradeable_packages, $upgradeable_packages)
   assert_type(Unattended_upgrades::Upgradeable_packages, $_upgradeable_packages)
 
   if $options != {} {
     warning('passing "options" is deprecated, use apt::conf directly instead')
   }
   $_options = merge($unattended_upgrades::default_options, $options)
   assert_type(Unattended_upgrades::Options, $_options)
 
   package { 'unattended-upgrades':
     ensure => $package_ensure,
   }
 
   apt::conf { 'unattended-upgrades':
     priority      => 50,
     content       => template("${module_name}/unattended-upgrades.erb"),
     require       => Package['unattended-upgrades'],
     notify_update => $notify_update,
   }
 
   apt::conf { 'periodic':
     priority      => 10,
     content       => template("${module_name}/periodic.erb"),
     require       => Package['unattended-upgrades'],
     notify_update => $notify_update,
   }
 
   apt::conf { 'auto-upgrades':
     ensure        => absent,
     priority      => 20,
     require       => Package['unattended-upgrades'],
     notify_update => $notify_update,
   }
   apt::conf { 'options':
     priority      => 10,
     content       => template("${module_name}/options.erb"),
     require       => Package['unattended-upgrades'],
     notify_update => $notify_update,
   }
 }
diff --git a/types/auto.pp b/types/auto.pp
index bc3a896..b37539e 100644
--- a/types/auto.pp
+++ b/types/auto.pp
@@ -1,9 +1,9 @@
 type Unattended_upgrades::Auto = Struct[
   {
-    Optional['clean']                => Integer[0],
+    Optional['clean']                => Variant[Integer[0], Enum['always']],
     Optional['fix_interrupted_dpkg'] => Boolean,
     Optional['reboot']               => Boolean,
     Optional['reboot_time']          => String,
     Optional['remove']               => Boolean,
   }
 ]
diff --git a/types/upgradeable_packages.pp b/types/upgradeable_packages.pp
index ce41a94..9fcd87e 100644
--- a/types/upgradeable_packages.pp
+++ b/types/upgradeable_packages.pp
@@ -1,6 +1,6 @@
 type Unattended_upgrades::Upgradeable_packages = Struct[
   {
-    Optional['download_only'] => Integer[0],
+    Optional['download_only'] => Variant[Integer[0], Enum['always']],
     Optional['debdelta']      => Integer[0],
   }
 ]