diff --git a/manifests/init.pp b/manifests/init.pp
index d04ffc0..c4a7555 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -1,6 +1,9 @@
 # == Class: unattended_upgrades
 #
-class unattended_upgrades($ensure = present) {
+class unattended_upgrades(
+  $ensure = present,
+  $origins = []
+) {
   package { 'unattended-upgrades':
     ensure => $ensure,
   }
@@ -23,7 +26,7 @@ class unattended_upgrades($ensure = present) {
     owner   => 'root',
     group   => 'root',
     mode    => '0444',
-    source  => 'puppet:///modules/unattended_upgrades/50unattended-upgrades',
+    content => template('unattended_upgrades/50unattended-upgrades.erb'),
     replace => true,
   }
 }
diff --git a/files/50unattended-upgrades b/templates/50unattended-upgrades.erb
similarity index 94%
rename from files/50unattended-upgrades
rename to templates/50unattended-upgrades.erb
index 5faa45e..2815768 100644
--- a/files/50unattended-upgrades
+++ b/templates/50unattended-upgrades.erb
@@ -5,6 +5,9 @@ Unattended-Upgrade::Allowed-Origins {
     "${distro_id} ${distro_codename}-security";
     "${distro_id} ${distro_codename}-updates";
 //  "${distro_id} ${distro_codename}-proposed-updates";
+<% origins.each do |origin| -%>
+    "<%= origin %>";
+<% end -%>
 };
 
 // List of packages to not update