From 1c1c47f5b2673c2239b7a2b1890a0d4aff0a18d6 Mon Sep 17 00:00:00 2001
From: Mark <mark@betalupi.com>
Date: Mon, 16 Mar 2026 22:24:39 -0700
Subject: [PATCH] Cancel fix

---
 crates/pile-toolbox/src/cancelabletask.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/crates/pile-toolbox/src/cancelabletask.rs b/crates/pile-toolbox/src/cancelabletask.rs
index d91f879..3ae4cf6 100644
--- a/crates/pile-toolbox/src/cancelabletask.rs
+++ b/crates/pile-toolbox/src/cancelabletask.rs
@@ -80,10 +80,12 @@ impl CancelFlag {
 
 	#[inline]
 	pub async fn await_cancel(&self) {
-		if self.is_cancelled() {
-			return;
+		let notified = self.notify.notified();
+		tokio::pin!(notified);
+		notified.as_mut().enable();
+		if !self.is_cancelled() {
+			notified.await;
 		}
-		self.notify.notified().await;
 	}
 
 	#[inline]