diff --git a/tests/milvus_memory_test.py b/tests/milvus_memory_test.py
index e0e2f7fc..84fd6e6d 100644
--- a/tests/milvus_memory_test.py
+++ b/tests/milvus_memory_test.py
@@ -26,7 +26,7 @@ try:
 
         def setUp(self) -> None:
             """Set up the test environment"""
-            self.cfg = MockConfig()
+            self.cfg = mock_config()
             self.memory = MilvusMemory(self.cfg)
 
         def test_add(self) -> None: