diff --git a/tasks/roles/setup_iptables/tasks/main.yml b/tasks/roles/setup_iptables/tasks/main.yml
index 4425155..4b91019 100644
--- a/tasks/roles/setup_iptables/tasks/main.yml
+++ b/tasks/roles/setup_iptables/tasks/main.yml
@@ -22,7 +22,16 @@
     state: directory
     mode: '0755'
 
-- name: Save rules to keep them on reboot
+- name: Install iptables-persistent for rules persistence
+  ansible.builtin.package:
+    name: iptables-persistent
+
+- name: Save IPv4 rules to keep them on reboot
   community.general.iptables_state:
     state: saved
-    path: /etc/iptables/iptables
\ No newline at end of file
+    path: /etc/iptables/rules.v4
+
+- name: Save IPv6 rules to keep them on reboot
+  community.general.iptables_state:
+    state: saved
+    path: /etc/iptables/rules.v6
\ No newline at end of file