From 6119cb1aaf8cb01a11834e2429cf0c0c9e4c0d33 Mon Sep 17 00:00:00 2001
From: cte <cestreich@gmail.com>
Date: Tue, 11 Feb 2025 12:48:19 -0800
Subject: [PATCH] Update comment

---
 src/core/Cline.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/core/Cline.ts b/src/core/Cline.ts
index f9a179736b..33865601aa 100644
--- a/src/core/Cline.ts
+++ b/src/core/Cline.ts
@@ -739,9 +739,9 @@ export class Cline {
 		this.urlContentFetcher.closeBrowser()
 		this.browserSession.closeBrowser()
 
+		// If we're not streaming then `abortStream` (which reverts the diff
+		// view changes) won't be called, so we need to revert the changes here.
 		if (this.isStreaming && this.diffViewProvider.isEditing) {
-			// Need to await for when we want to make sure directories/files are
-			// reverted before re-starting the task from a checkpoint.
 			await this.diffViewProvider.revertChanges()
 		}
 	}