diff --git a/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp b/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp
index 52770062..1c3099bd 100644
--- a/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp
+++ b/global/overlay/etc/puppet/modules/eid/manifests/telegraf2.pp
@@ -6,14 +6,15 @@
 # @param service_dir          The directory where all related config and files are stored. (referenced in compose file)
 # @param influx_urls          List of influxdb urls to send data to
 class eid::telegraf2 (
-  String $version                = 'latest',
-  String $server_fqdn            = $facts['networking']['fqdn'],
-  Array  $ports                  = [],
-  String $service_dir            = '/etc/telegraf',
-  Array  $influx_urls            = ["https://metrics-external-1.sunet.se:8086"],
-  String $influxdb2_org          = 'SwedenConnect',
-  String $influxdb2_bucket       = 'eidas',
-  String $influxdb2_bucket_proxy = 'eidas-proxy',
+  String $version                    = 'latest',
+  String $server_fqdn                = $facts['networking']['fqdn'],
+  Enum['proxy', 'log'] $server_type,
+  Array  $ports                      = [],
+  String $service_dir                = '/etc/telegraf',
+  Array  $influx_urls                = ["https://metrics-external-1.sunet.se:8086"],
+  String $influxdb2_org              = 'SwedenConnect',
+  String $influxdb2_bucket           = 'eidas',
+  String $influxdb2_bucket_proxy     = 'eidas-proxy',
 ) {
 
   # Get influx password token from eyaml