diff --git a/states/nftables/config.sls b/states/nftables/config.sls
index e847a7f..c6c0d87 100644
--- a/states/nftables/config.sls
+++ b/states/nftables/config.sls
@@ -7,7 +7,7 @@ nftables-config-dir:
 nftables-blacklist-config:
   file.managed:
     - name: {{ nftables.config_dir }}/blacklist.nft
-    - source: salt://nftables/blacklist.nft.j2
+    - source: salt://nftables/templates/blacklist.nft.j2
     - template: jinja
     - watch_in:
       - service: nftables-service
@@ -18,7 +18,7 @@ nftables-blacklist-config:
 nftables-rules-config:
   file.managed:
     - name: {{ nftables.config_dir }}/rules.nft
-    - source: salt://nftables/rules.nft.j2
+    - source: salt://nftables/templates/rules.nft.j2
     - template: jinja
     - watch_in:
       - service: nftables-service
@@ -29,7 +29,7 @@ nftables-rules-config:
 nftables-main-config:
   file.managed:
     - name: /etc/nftables.conf
-    - source: salt://nftables/nftables.conf.j2
+    - source: salt://nftables/templates/nftables.conf.j2
     - template: jinja
     - watch_in:
       - service: nftables-service
diff --git a/states/nftables/rules.nft.j2 b/states/nftables/rules.nft.j2
deleted file mode 100644
index b5cbfcb..0000000
--- a/states/nftables/rules.nft.j2
+++ /dev/null
@@ -1,51 +0,0 @@
-## {{ salt['pillar.get']('salt_managed', default='Salt Managed') }}
-
-{%- from "nftables/map.jinja" import nftables with context %}
-{%- from "nftables/map.jinja" import net with context %}
-
-## IPv4 filtering
-add table ip filter
-add chain ip filter input { type filter hook input priority 0; policy drop; }
-add chain ip filter forward { type filter hook forward priority 0; policy accept; }
-add chain ip filter output { type filter hook output priority 0; policy accept; }
-add rule ip filter input iifname lo counter accept
-add rule ip filter input iifname tun* counter accept
-add rule ip filter input ct state related,established counter accept
-add rule ip filter input ip protocol icmp counter accept
-add rule ip filter input ip saddr $blacklist drop
-{%- for key, value in net.ipv4_networks.items() %}
-add rule ip filter input ip saddr {{ value.ip }}/{{ value.mask }} ct state established,new counter accept
-{%- endfor %}
-{%- for key, value in net.public_ports.items() %}
-add rule ip filter input {{ value.proto }} dport {{ value.port }} ct state established,new counter accept
-{%- endfor %}
-#add rule ip filter input counter log
-
-## IPv4 NAT
-add table ip nat
-add chain ip nat prerouting { type nat hook prerouting priority 0; policy accept; }
-add chain ip nat input { type nat hook input priority 0; policy accept; }
-add chain ip nat output { type nat hook output priority 0; policy accept; }
-add chain ip nat postrouting { type nat hook postrouting priority 0; policy accept; }
-{%- for key, value in net.nats.items() %}
-add rule ip nat postrouting ip saddr {{ value.ip }}/{{ value.mask }} counter masquerade
-{%- endfor %}
-
-## IPv6 filtering
-add table ip6 filter6
-add chain ip6 filter6 input { type filter hook input priority 0; policy drop; }
-add chain ip6 filter6 forward { type filter hook forward priority 0; policy accept; }
-add chain ip6 filter6 output { type filter hook output priority 0; policy accept; }
-add rule ip6 filter6 input iifname lo counter accept
-add rule ip6 filter6 input iifname tun* counter accept
-add rule ip6 filter6 input ct state related,established counter accept
-add rule ip6 filter6 input icmpv6 type {destination-unreachable, packet-too-big, time-exceeded, echo-request, echo-reply, mld-listener-query, mld-listener-report, mld-listener-reduction, nd-router-solicit, nd-router-advert, nd-neighbor-solicit, nd-neighbor-advert, nd-redirect, parameter-problem, router-renumbering} accept
-{%- for key, value in net.ipv6_networks.items() %}
-add rule ip6 filter6 input ip6 saddr {{ value.ip }}/{{ value.mask }} ct state established,new counter accept
-{%- endfor %}
-{%- for key, value in net.public_ports.items() %}
-add rule ip6 filter6 input {{ value.proto }} dport {{ value.port }} ct state established,new counter accept
-{%- endfor %}
-#add rule ip6 filter6 input counter log
-
-## Endline is mandatory
diff --git a/states/nftables/blacklist.nft.j2 b/states/nftables/templates/blacklist.nft.j2
similarity index 100%
rename from states/nftables/blacklist.nft.j2
rename to states/nftables/templates/blacklist.nft.j2
diff --git a/states/nftables/nftables.conf.j2 b/states/nftables/templates/nftables.conf.j2
similarity index 100%
rename from states/nftables/nftables.conf.j2
rename to states/nftables/templates/nftables.conf.j2
diff --git a/states/nftables/templates/rules.nft.j2 b/states/nftables/templates/rules.nft.j2
new file mode 100644
index 0000000..f20ff04
--- /dev/null
+++ b/states/nftables/templates/rules.nft.j2
@@ -0,0 +1,51 @@
+## {{ salt['pillar.get']('salt_managed', default='Salt Managed') }}
+
+{%- from "nftables/map.jinja" import nftables with context %}
+{%- from "nftables/map.jinja" import net with context %}
+
+## IPv4 filtering
+add table ip filter
+add chain ip filter INPUT { type filter hook input priority 0; policy drop; }
+add chain ip filter FORWARD { type filter hook forward priority 0; policy accept; }
+add chain ip filter OUTPUT { type filter hook output priority 0; policy accept; }
+add rule ip filter INPUT iifname lo counter accept
+add rule ip filter INPUT iifname tun* counter accept
+add rule ip filter INPUT ct state related,established counter accept
+add rule ip filter INPUT ip protocol icmp counter accept
+add rule ip filter INPUT log ip saddr $blacklist drop
+{%- for key, value in net.ipv4_networks.items() %}
+add rule ip filter INPUT ip saddr {{ value.ip }}/{{ value.mask }} ct state established,new counter accept
+{%- endfor %}
+{%- for key, value in net.public_ports.items() %}
+add rule ip filter INPUT {{ value.proto }} dport {{ value.port }} ct state established,new counter accept
+{%- endfor %}
+#add rule ip filter INPUT counter log
+
+## IPv4 NAT
+add table ip nat
+add chain ip nat PREROUTING { type nat hook prerouting priority 0; policy accept; }
+add chain ip nat INPUT { type nat hook input priority 0; policy accept; }
+add chain ip nat OUTPUT { type nat hook output priority 0; policy accept; }
+add chain ip nat POSTROUTING { type nat hook postrouting priority 0; policy accept; }
+{%- for key, value in net.nats.items() %}
+add rule ip nat POSTROUTING ip saddr {{ value.ip }}/{{ value.mask }} counter masquerade
+{%- endfor %}
+
+## IPv6 filtering
+add table ip6 filter6
+add chain ip6 filter6 INPUT { type filter hook input priority 0; policy drop; }
+add chain ip6 filter6 FORWARD { type filter hook forward priority 0; policy accept; }
+add chain ip6 filter6 OUTPUT { type filter hook output priority 0; policy accept; }
+add rule ip6 filter6 INPUT iifname lo counter accept
+add rule ip6 filter6 INPUT iifname tun* counter accept
+add rule ip6 filter6 INPUT ct state related,established counter accept
+add rule ip6 filter6 INPUT icmpv6 type {destination-unreachable, packet-too-big, time-exceeded, echo-request, echo-reply, mld-listener-query, mld-listener-report, mld-listener-reduction, nd-router-solicit, nd-router-advert, nd-neighbor-solicit, nd-neighbor-advert, nd-redirect, parameter-problem, router-renumbering} accept
+{%- for key, value in net.ipv6_networks.items() %}
+add rule ip6 filter6 INPUT ip6 saddr {{ value.ip }}/{{ value.mask }} ct state established,new counter accept
+{%- endfor %}
+{%- for key, value in net.public_ports.items() %}
+add rule ip6 filter6 INPUT {{ value.proto }} dport {{ value.port }} ct state established,new counter accept
+{%- endfor %}
+#add rule ip6 filter6 INPUT counter log
+
+## Endline is mandatory