diff --git a/autogpt/memory/milvus.py b/autogpt/memory/milvus.py
index fc450196b524..64323df93180 100644
--- a/autogpt/memory/milvus.py
+++ b/autogpt/memory/milvus.py
@@ -31,7 +31,7 @@ def __init__(self, cfg: Config) -> None:
             secure=self.secure,
         )
 
-        self.init_collection()
+        self.init_collection(cfg)
 
     def configure(self, cfg: Config) -> None:
         # init with configuration.
@@ -69,7 +69,7 @@ def configure(self, cfg: Config) -> None:
                 "params": {},
             }
 
-    def init_collection(self) -> None:
+    def init_collection(self, cfg: Config) -> None:
         """Initialize collection in vector database."""
         fields = [
             FieldSchema(name="pk", dtype=DataType.INT64, is_primary=True, auto_id=True),