diff --git a/yunohost_firewall.py b/yunohost_firewall.py
index 13413111..78dee3b4 100644
--- a/yunohost_firewall.py
+++ b/yunohost_firewall.py
@@ -37,6 +37,7 @@ except ImportError:
     sys.stderr.write('apt-get install python-yaml\n')
     sys.exit(1)
 from yunohost import YunoHostError, win_msg
+from yunohost_hook import hook_callback
 
 
 def firewall_allow(protocol=None, port=None, ipv6=None, upnp=False):