diff --git a/tests/milvus_memory_test.py b/tests/milvus_memory_test.py
index 46010f7e..9672326b 100644
--- a/tests/milvus_memory_test.py
+++ b/tests/milvus_memory_test.py
@@ -5,22 +5,21 @@ import sys
 import unittest
 
 try:
+    from autogpt.config import Config
     from autogpt.memory.milvus import MilvusMemory
 
-    def mock_config() -> dict:
+    def mock_config() -> Config:
         """Mock the config object for testing purposes."""
+
         # Return a mock config object with the required attributes
-        return type(
-            "MockConfig",
-            (object,),
-            {
-                "debug_mode": False,
-                "continuous_mode": False,
-                "speak_mode": False,
-                "milvus_collection": "autogpt",
-                "milvus_addr": "localhost:19530",
-            },
-        )
+        class MockConfig(Config):
+            debug_mode = False
+            continuous_mode = False
+            speak_mode = False
+            milvus_collection = "autogpt"
+            milvus_addr = "localhost:19530"
+
+        return MockConfig()
 
     class TestMilvusMemory(unittest.TestCase):
         """Tests for the MilvusMemory class."""