diff --git a/working_dir_error_replication/reader.py b/working_dir_error_replication/reader.py
index 5c54b771b..4ef78ea06 100644
--- a/working_dir_error_replication/reader.py
+++ b/working_dir_error_replication/reader.py
@@ -6,7 +6,6 @@ from cognee.infrastructure.databases.graph.kuzu.adapter import KuzuAdapter
 
 
 async def main():
-    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 fb358b886..07f2cd276 100644
--- a/working_dir_error_replication/writer.py
+++ b/working_dir_error_replication/writer.py
@@ -20,14 +20,14 @@ def create_node(name):
 
 async def main():
     adapter = KuzuAdapter("test.db")
-    nodes = [create_node(f"Node{i}") for i in range(200000)]
+    nodes = [create_node(f"Node{i}") for i in range(2)]
 
     print("Writer: Starting...")
     await adapter.add_nodes(nodes)
 
     print("writer finished...")
 
-    time.sleep(10)
+    time.sleep(5)
 
 
 if __name__ == "__main__":