From 06283f0ffb0d9531815dadf049ac3578d4430e97 Mon Sep 17 00:00:00 2001
From: Kochoyume <78918084+kochounoyume@users.noreply.github.com>
Date: Tue, 24 Sep 2024 23:58:28 +0900
Subject: [PATCH] Add to the code that I forgot to write

---
 .../Assets/Plugins/UniTask/Runtime/UniTask.WaitUntil.cs  | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.WaitUntil.cs b/src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.WaitUntil.cs
index 2f29b64..b113353 100644
--- a/src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.WaitUntil.cs
+++ b/src/UniTask/Assets/Plugins/UniTask/Runtime/UniTask.WaitUntil.cs
@@ -241,6 +241,10 @@ namespace Cysharp.Threading.Tasks
                     {
                         TryReturn();
                     }
+                    else
+                    {
+                        TaskTracker.RemoveTracking(this);
+                    }
                 }
             }
 
@@ -461,6 +465,7 @@ namespace Cysharp.Threading.Tasks
                 result.predicate = predicate;
                 result.argument = argument;
                 result.cancellationToken = cancellationToken;
+                result.cancelImmediately = cancelImmediately;
 
                 if (cancelImmediately && cancellationToken.CanBeCanceled)
                 {
@@ -491,6 +496,10 @@ namespace Cysharp.Threading.Tasks
                     {
                         TryReturn();
                     }
+                    else
+                    {
+                        TaskTracker.RemoveTracking(this);
+                    }
                 }
             }