3 files changed, 20 insertions, 14 deletions
diff --git a/manifests/virtual/munin.pp b/manifests/virtual/munin.pp
index 531acd9..3db8b7e 100644
--- a/manifests/virtual/munin.pp
+++ b/manifests/virtual/munin.pp
@@ -6,11 +6,13 @@ define firewall::virtual::munin(
   $zone           = 'fw'
   $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
 ) {
-  class { "firewall::implementations::${implementation}::virtual::munin":
-    destination => $destination,
-    port_orig   => $port_orig,
-    port_dest   => $port_dest,
-    order       => $order,
-    zone        => $zone,
+  if $implementation == 'shorewall' {
+    firewall::implementations::shorewall::virtual::munin { "${name}":
+      destination => $destination,
+      port_orig   => $port_orig,
+      port_dest   => $port_dest,
+      order       => $order,
+      zone        => $zone,
+    }
   }
 }
diff --git a/manifests/virtual/ssh.pp b/manifests/virtual/ssh.pp
index 0a11fa2..98fa976 100644
--- a/manifests/virtual/ssh.pp
+++ b/manifests/virtual/ssh.pp
@@ -2,13 +2,15 @@ define firewall::virtual::ssh(
   $destination,
   $port_orig      = '22',
   $port_dest      = '',
-  $zone           = 'vm'
+  $zone           = 'vm',
   $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
 ) {
-  class { "firewall::implementations::${implementation}::virtual::ssh":
-    destination => $destination,
-    port_orig   => $port_orig,
-    port_dest   => $port_dest,
-    zone        => $zone,
+  if $implementation == 'shorewall' {
+    firewall::implementations::shorewall::virtual::ssh { "${name}":
+      destination => $destination,
+      port_orig   => $port_orig,
+      port_dest   => $port_dest,
+      zone        => $zone,
+    }
   }
 }
diff --git a/manifests/virtual/web.pp b/manifests/virtual/web.pp
index c54a95b..d551d4b 100644
--- a/manifests/virtual/web.pp
+++ b/manifests/virtual/web.pp
@@ -2,7 +2,9 @@ define firewall::virtual::web(
   $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
   $destination
 ) {
-  class { "firewall::implementations::${implementation}::virtual::web":
-    destination => $destination,
+  if $implementation == 'shorewall' {
+    firewall::implementations::shorewall::virtual::web { "${name}":
+      destination => $destination,
+    }
   }
 }