diff --git a/global/overlay/etc/puppet/modules/net/manifests/nftables.pp b/global/overlay/etc/puppet/modules/net/manifests/nftables.pp
index c7bc2c9..ff06af8 100644
--- a/global/overlay/etc/puppet/modules/net/manifests/nftables.pp
+++ b/global/overlay/etc/puppet/modules/net/manifests/nftables.pp
@@ -1,8 +1,10 @@
-$nftables_rules = lookup('sunet::nftables::allow::rules', Hash, 'deep', {})
+class net::nftables {
+  $nftables_rules = lookup('sunet::nftables::allow::rules', Hash, 'deep', {})
 
-# Create resources for each rule
-$nftables_rules.each |$name, $params| {
-  sunet::nftables::allow { $name:
-    * => $params,
+  $nftables_rules.each |$name, $params| {
+    sunet::nftables::allow { $name:
+      * => $params,
+    }
   }
+
 }