From 1693d5f9199491167b9b2650615a0f620b669eb9 Mon Sep 17 00:00:00 2001
From: Anairkoen Schno <aaron.cirr.com@gmail.com>
Date: Tue, 4 Feb 2020 15:00:58 -0600
Subject: [PATCH] Swapped updater coroutine waiter for utility function

---
 IPA.Loader/Updating/BeatMods/Updater.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/IPA.Loader/Updating/BeatMods/Updater.cs b/IPA.Loader/Updating/BeatMods/Updater.cs
index 37909668..200c8d0c 100644
--- a/IPA.Loader/Updating/BeatMods/Updater.cs
+++ b/IPA.Loader/Updating/BeatMods/Updater.cs
@@ -14,6 +14,7 @@ using IPA.Config;
 using IPA.Loader;
 using IPA.Loader.Features;
 using IPA.Utilities;
+using IPA.Utilities.Async;
 using Newtonsoft.Json;
 using SemVer;
 using UnityEngine;
@@ -527,8 +528,7 @@ namespace IPA.Updating.BeatMods
                         ExtractPluginAsync(stream, item, platformFile);
                     }, taskTokenSource.Token);
 
-                    while (!(downloadTask.IsCompleted || downloadTask.IsCanceled || downloadTask.IsFaulted))
-                        yield return null; // pause co-routine until task is done
+                    yield return Coroutines.WaitForTask(downloadTask);
 
                     if (downloadTask.IsFaulted)
                     {