1 files changed, 8 insertions, 7 deletions
diff --git a/manifests/router/ssh.pp b/manifests/router/ssh.pp
index 6f1a640..5704fbb 100644
--- a/manifests/router/ssh.pp
+++ b/manifests/router/ssh.pp
@@ -6,12 +6,13 @@ define firewall::router::ssh(
   $originaldest   = $ipaddress,
   $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
 ) {
-  class { "firewall::implementations::${implementation}::router::ssh":
-    destination  => $destination,
-    port_orig    => $port_orig,
-    port_dest    => $port_dest,
-    zone         => $zone,
-    order        => $order,
-    originaldest => $originaldest,
+  if $implementation == 'shorewall' {
+    firewall::implementations::shorewall::router::ssh { "${name}":
+      destination  => $destination,
+      port_orig    => $port_orig,
+      port_dest    => $port_dest,
+      zone         => $zone,
+      originaldest => $originaldest,
+    }
   }
 }