diff --git a/pkg/iptables/iptables.go b/pkg/iptables/iptables.go
index c3308d0..1c0a399 100644
--- a/pkg/iptables/iptables.go
+++ b/pkg/iptables/iptables.go
@@ -380,7 +380,7 @@ func (c *Controller) Set(rules []Rule) error {
 			}
 
 			level.Debug(c.logger).Log("msg", "Applying Firewall Rule...", "Rule", ruleString, "Protocol", protocolName)
-			if err := rules[i].Add(c.client(proto)); err != nil {
+			if err := rules[i].Add(c.v4); err != nil {
 				return fmt.Errorf("failed to add rule: %v", err)
 			}
 			level.Debug(c.logger).Log("msg", "Firewall Rule applied.", "Rule", ruleString, "Protocol", protocolName)