1 files changed, 12 insertions, 9 deletions
diff --git a/manifests/firewall.pp b/manifests/firewall.pp
index 10f5988..124bb07 100644
--- a/manifests/firewall.pp
+++ b/manifests/firewall.pp
@@ -1,12 +1,15 @@
-import "config.pp"
 import "shorewall"
 
-shorewall::rule { "ssh-02":
-  action          => 'DNAT',
-  source          => 'net',
-  destination     => "vm:192.168.0.$context:22",
-  proto           => 'tcp',
-  destinationport => "22$id",
-  ratelimit       => '-',
-  order           => "202",
+class firewall {
+  include shorewall
+
+  shorewall::rule { "ssh-02":
+    action          => 'DNAT',
+    source          => 'net',
+    destination     => "vm:192.168.0.$context:22",
+    proto           => 'tcp',
+    destinationport => "22$id",
+    ratelimit       => '-',
+    order           => "202",
+  }
 }