diff --git a/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py b/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py
index 6947a8601819..e5aa2a2baa77 100644
--- a/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py
+++ b/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py
@@ -52,3 +52,12 @@ def __repr__(self):
 
     def __str__(self):
         return repr(self.fv_dict)
+
+class ConfigDBConnector:
+    pass
+
+class SonicDBConfig:
+    pass
+
+class SonicV2Connector:
+    pass
diff --git a/sonic-pcied/tests/test_DaemonPcied.py b/sonic-pcied/tests/test_DaemonPcied.py
index c8c17a276edb..2c3c953e7e48 100644
--- a/sonic-pcied/tests/test_DaemonPcied.py
+++ b/sonic-pcied/tests/test_DaemonPcied.py
@@ -11,14 +11,11 @@
 else:
     import mock
 
-from sonic_py_common import daemon_base
-
 from .mock_platform import MockPcieUtil
 
 SYSLOG_IDENTIFIER = 'pcied_test'
 NOT_AVAILABLE = 'N/A'
 
-daemon_base.db_connect = mock.MagicMock()
 
 tests_path = os.path.dirname(os.path.abspath(__file__))
 
@@ -26,6 +23,9 @@
 mocked_libs_path = os.path.join(tests_path, "mocked_libs")
 sys.path.insert(0, mocked_libs_path)
 
+from sonic_py_common import daemon_base
+daemon_base.db_connect = mock.MagicMock()
+
 # Add path to the file under test so that we can load it
 modules_path = os.path.dirname(tests_path)
 scripts_path = os.path.join(modules_path, "scripts")
diff --git a/sonic-pcied/tests/test_pcied.py b/sonic-pcied/tests/test_pcied.py
index a99331846997..3850d38ed156 100644
--- a/sonic-pcied/tests/test_pcied.py
+++ b/sonic-pcied/tests/test_pcied.py
@@ -10,7 +10,6 @@
 else:
     from mock import MagicMock, patch, mock_open
 
-from sonic_py_common import daemon_base, device_info
 from .mock_platform import MockPcieUtil
 
 tests_path = os.path.dirname(os.path.abspath(__file__))
@@ -18,6 +17,7 @@
 # Add mocked_libs path so that the file under test can load mocked modules from there
 mocked_libs_path = os.path.join(tests_path, "mocked_libs")
 sys.path.insert(0, mocked_libs_path)
+from sonic_py_common import daemon_base, device_info
 
 # Add path to the file under test so that we can load it
 modules_path = os.path.dirname(tests_path)