diff --git a/src/tests/helper.py b/src/tests/helper.py
index 636e1ec..c3aed20 100644
--- a/src/tests/helper.py
+++ b/src/tests/helper.py
@@ -18,6 +18,7 @@
 import importlib.util
 import math
 import json
+import os
 
 def prepare_module(file_path='', module_name=''):
     spec = importlib.util.spec_from_file_location(module_name, file_path)
diff --git a/src/tests/test_initial_setup.py b/src/tests/test_initial_setup.py
index f85bf12..4a0f371 100644
--- a/src/tests/test_initial_setup.py
+++ b/src/tests/test_initial_setup.py
@@ -17,6 +17,7 @@
 import unittest
 import vyos.configtree
 import vyos.initialsetup as vis
+import json
 
 from unittest import TestCase
 from vyos.xml_ref import definition