diff --git a/working_dir_error_replication/reader.py b/working_dir_error_replication/reader.py
index b81857ade..caf8407c1 100644
--- a/working_dir_error_replication/reader.py
+++ b/working_dir_error_replication/reader.py
@@ -6,9 +6,7 @@ from cognee.infrastructure.databases.graph.kuzu.adapter import KuzuAdapter
 
 
 async def main():
-    print("Reader: Waiting 2 seconds...")
-
-    time.sleep(5)
+    time.sleep(20)
     adapter = KuzuAdapter("test.db")
     result = await adapter.query("MATCH (n:Node) RETURN COUNT(n)")
 
diff --git a/working_dir_error_replication/writer.py b/working_dir_error_replication/writer.py
index 2e0ce1728..928ba4d2a 100644
--- a/working_dir_error_replication/writer.py
+++ b/working_dir_error_replication/writer.py
@@ -19,13 +19,14 @@ def create_node(name):
 
 async def main():
     adapter = KuzuAdapter("test.db")
+    nodes = [create_node(f"Node{i}") for i in range(200000)]
+
     print("Writer: Starting...")
-    nodes = [create_node(f"Node{i}") for i in range(100)]
     await adapter.add_nodes(nodes)
 
     print("writer finished...")
 
-    time.sleep(100)
+    time.sleep(10)
 
 if __name__ == "__main__":
     asyncio.run(main())
\ No newline at end of file