diff --git a/global/overlay/etc/puppet/modules/net/manifests/baas2_restoretest.pp b/global/overlay/etc/puppet/modules/net/manifests/baas2_restoretest.pp
index 97a3279..cc55f0f 100644
--- a/global/overlay/etc/puppet/modules/net/manifests/baas2_restoretest.pp
+++ b/global/overlay/etc/puppet/modules/net/manifests/baas2_restoretest.pp
@@ -1,7 +1,9 @@
 # This puppet manifest is used to test backup and restore in an automated fashion
 # This tests the backup and restores when manged by sunet::baas2 from puppet-sunet
 # Jira-ref: SUNETOPS-1997
-class net::baas2_restoretest() {
+class net::baas2_restoretest(
+  Enum['backupnode', 'restorenode'] $node_type = undef,
+) {
 
   file { '/opt/restore-target':
     ensure => directory,