1 files changed, 6 insertions, 4 deletions
diff --git a/manifests/router/tor.pp b/manifests/router/tor.pp
index b93ea05..e3c61fb 100644
--- a/manifests/router/tor.pp
+++ b/manifests/router/tor.pp
@@ -2,9 +2,11 @@ define firewall::router::tor(
   $implementation = lookup('firewall::implementation', undef, undef, 'shorewall'),
   $destination, $zone = 'loc', $originaldest = $ipaddress
 ) {
-  class { "firewall::implementations::${implementation}::router::tor":
-    destination  => $destination,
-    zone         => $zone,
-    originaldest => $originaldest,
+  if $implementation == 'shorewall' {
+    firewall::implementations::shorewall::router::tor { "${name}":
+      destination  => $destination,
+      zone         => $zone,
+      originaldest => $originaldest,
+    }
   }
 }