diff --git a/src/agent/agent.js b/src/agent/agent.js
index bfa9c53..acd947b 100644
--- a/src/agent/agent.js
+++ b/src/agent/agent.js
@@ -485,6 +485,7 @@ export class Agent {
             if (res) {
                 await this.history.add('system', `Task ended with score : ${res.score}`);
                 await this.history.save();
+                await new Promise(resolve => setTimeout(resolve, 3000)); // Wait 3 second for save to complete
                 console.log('Task finished:', res.message);
                 this.killAll();
             }