diff --git a/hooks/neutron_ovs_utils.py b/hooks/neutron_ovs_utils.py
index 55a8c8d7..252656a0 100644
--- a/hooks/neutron_ovs_utils.py
+++ b/hooks/neutron_ovs_utils.py
@@ -116,5 +116,6 @@ def get_shared_secret():
     if 'shared_secret' in ctxt:
         return ctxt['shared_secret']
 
+
 def use_dvr():
     return neutron_ovs_context.use_dvr()
diff --git a/unit_tests/test_neutron_ovs_hooks.py b/unit_tests/test_neutron_ovs_hooks.py
index c4180a98..dfa7d29d 100644
--- a/unit_tests/test_neutron_ovs_hooks.py
+++ b/unit_tests/test_neutron_ovs_hooks.py
@@ -21,6 +21,7 @@ utils.restart_map = _map
 TO_PATCH = [
     'apt_update',
     'apt_install',
+    'apt_purge',
     'config',
     'CONFIGS',
     'determine_packages',
@@ -30,6 +31,7 @@ TO_PATCH = [
     'relation_ids',
     'relation_set',
     'configure_ovs',
+    'use_dvr',
 ]
 NEUTRON_CONF_DIR = "/etc/neutron"