diff --git a/python/samples/concepts/memory/new_memory.py b/python/samples/concepts/memory/new_memory.py
index 91f4a2d8cd6b..559416ea4d5e 100644
--- a/python/samples/concepts/memory/new_memory.py
+++ b/python/samples/concepts/memory/new_memory.py
@@ -269,7 +269,7 @@ async def main(collection: str, use_azure_openai: bool, embedding_model: str):
     argparse.ArgumentParser()
 
     parser = argparse.ArgumentParser()
-    parser.add_argument("--collection", default="weaviate", choices=collections.keys(), help="What collection to use.")
+    parser.add_argument("--collection", default="in_memory", choices=collections.keys(), help="What collection to use.")
     # Option of whether to use OpenAI or Azure OpenAI.
     parser.add_argument("--use-azure-openai", action="store_true", help="Use Azure OpenAI instead of OpenAI.")
     # Model