diff --git a/lib/strategies/prune-tool.ts b/lib/strategies/prune-tool.ts
index 83d4ff5..1fccb44 100644
--- a/lib/strategies/prune-tool.ts
+++ b/lib/strategies/prune-tool.ts
@@ -73,9 +73,6 @@ export function createPruneTool(
             const pruneToolIds: string[] = getPruneToolIds(numericToolIds, toolIdList)
             state.prune.toolIds.push(...pruneToolIds)
 
-            saveSessionState(state, logger)
-                .catch(err => logger.error("Failed to persist state", { error: err.message }))
-
             const toolMetadata = new Map<string, ToolParameterEntry>()
             for (const id of pruneToolIds) {
                 const toolParameters = state.toolParameters.get(id)
@@ -104,6 +101,9 @@ export function createPruneTool(
             state.stats.pruneTokenCounter = 0
             state.nudgeCounter = 0
 
+            saveSessionState(state, logger)
+                .catch(err => logger.error("Failed to persist state", { error: err.message }))
+
             return formatPruningResultForTool(
                 pruneToolIds,
                 toolMetadata,