1 files changed, 12 insertions, 11 deletions
diff --git a/src/op_mode/dns_forwarding_reset.py b/src/op_mode/dns_forwarding_reset.py
index 93c2444b9..8e2ee546c 100755
--- a/src/op_mode/dns_forwarding_reset.py
+++ b/src/op_mode/dns_forwarding_reset.py
@@ -21,12 +21,11 @@
 
 
 import os
-import sys
 import argparse
 
-import vyos.config
-from vyos.util import run
-
+from sys import exit
+from vyos.config import Config
+from vyos.util import call
 
 parser = argparse.ArgumentParser()
 parser.add_argument("-a", "--all", action="store_true", help="Reset all cache")
@@ -36,16 +35,18 @@ if __name__ == '__main__':
     args = parser.parse_args()
 
     # Do nothing if service is not configured
-    c = vyos.config.Config()
-    if not c.exists_effective('service dns forwarding'):
+    c = Config()
+    if not c.exists_effective(['service', 'dns', 'forwarding']):
         print("DNS forwarding is not configured")
-        sys.exit(0)
+        exit(0)
 
     if args.all:
-        run("rec_control wipe-cache \'.$\'")
-        sys.exit(1)
+        call("rec_control wipe-cache \'.$\'")
+        exit(0)
+
     elif args.domain:
-        run("rec_control wipe-cache \'{0}$\'".format(args.domain))
+        call("rec_control wipe-cache \'{0}$\'".format(args.domain))
+
     else:
         parser.print_help()
-        sys.exit(1)
+        exit(1)