diff --git a/src/sonic-bgpcfgd/tests/test_config.py b/src/sonic-bgpcfgd/tests/test_config.py
index c8bd0c9ca2be..b7ba2682ea26 100644
--- a/src/sonic-bgpcfgd/tests/test_config.py
+++ b/src/sonic-bgpcfgd/tests/test_config.py
@@ -1,4 +1,5 @@
-from unittest.mock import MagicMock
+from mock import MagicMock
+
 
 from bgpcfgd.config import ConfigMgr