diff --git a/tests/client/test_firewall.py b/tests/client/test_firewall.py
index f82ddfa..e714c81 100644
--- a/tests/client/test_firewall.py
+++ b/tests/client/test_firewall.py
@@ -157,6 +157,7 @@ def test_main(mock_get_method, mock_setup_daemon, mock_rewrite_etc_hosts):
             None,
             None,
             '0x01'),
+        call().wait_for_firewall_ready(),
         call().restore_firewall(1024, AF_INET6, True, None, None),
         call().restore_firewall(1025, AF_INET, True, None, None),
     ]