diff --git a/test/unit/api_spec.js b/test/unit/api_spec.js
index 3a5c77ba9..cb0dcf9fd 100644
--- a/test/unit/api_spec.js
+++ b/test/unit/api_spec.js
@@ -1906,7 +1906,11 @@ describe("api", function () {
               viewport,
             });
 
-            pdfDoc
+            renderTask.onContinue = function (cont) {
+              waitSome(cont);
+            };
+
+            return pdfDoc
               .cleanup()
               .then(
                 () => {