diff --git a/lib/messages/prune.ts b/lib/messages/prune.ts
index 33d9a7a..fef5635 100644
--- a/lib/messages/prune.ts
+++ b/lib/messages/prune.ts
@@ -151,6 +151,11 @@ const pruneToolInputs = (
             if (part.tool !== 'write' && part.tool !== 'edit') {
                 continue
             }
+            // Don't prune yet if tool is still pending or running
+            if (part.state.status === "pending" || part.state.status === "running") {
+                continue
+            }
+
             if (part.state.input?.content !== undefined) {
                 part.state.input.content = PRUNED_TOOL_INPUT_REPLACEMENT
             }
diff --git a/package-lock.json b/package-lock.json
index e1c7021..42b0ebd 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@tarquinen/opencode-dcp",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@tarquinen/opencode-dcp",
-      "version": "1.0.3",
+      "version": "1.0.4",
       "license": "MIT",
       "dependencies": {
         "@ai-sdk/openai-compatible": "^1.0.28",
diff --git a/package.json b/package.json
index 9bcf62d..3732f49 100644
--- a/package.json
+++ b/package.json
@@ -1,7 +1,7 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
   "name": "@tarquinen/opencode-dcp",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "type": "module",
   "description": "OpenCode plugin that optimizes token usage by pruning obsolete tool outputs from conversation context",
   "main": "./dist/index.js",