diff --git a/app/lib/stores/workbench.ts b/app/lib/stores/workbench.ts
index 3e9806c84..1dbfc5b5b 100644
--- a/app/lib/stores/workbench.ts
+++ b/app/lib/stores/workbench.ts
@@ -309,7 +309,7 @@ export class WorkbenchStore {
 
       if (!isStreaming) {
         await artifact.runner.runAction(data);
-        this.resetCurrentDocument();
+        this.resetAllFileModifications();
       }
     } else {
       await artifact.runner.runAction(data);