Fix firewall class