From abe579d3321e69fa60043224ba8459b69d674ce8 Mon Sep 17 00:00:00 2001
From: Nick O'Leary <nick.oleary@gmail.com>
Date: Thu, 21 Mar 2024 19:27:37 +0000
Subject: [PATCH] Fix async tidy up handling in core tests

---
 test/nodes/core/common/20-inject_spec.js     | 2 +-
 test/nodes/core/network/22-websocket_spec.js | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/nodes/core/common/20-inject_spec.js b/test/nodes/core/common/20-inject_spec.js
index 056ca9693..9dba5098e 100644
--- a/test/nodes/core/common/20-inject_spec.js
+++ b/test/nodes/core/common/20-inject_spec.js
@@ -43,7 +43,7 @@ describe('inject node', function() {
         });
     }
 
-    afterEach(async function() {
+    afterEach(function(done) {
         helper.unload().then(function () {
             return Context.clean({allNodes: {}});
         }).then(function () {
diff --git a/test/nodes/core/network/22-websocket_spec.js b/test/nodes/core/network/22-websocket_spec.js
index 21f415531..ed561207b 100644
--- a/test/nodes/core/network/22-websocket_spec.js
+++ b/test/nodes/core/network/22-websocket_spec.js
@@ -66,7 +66,7 @@ describe('websocket Node', function() {
 
     afterEach(function() {
         closeAll();
-        helper.unload();
+        return helper.unload();
     });
 
     describe('websocket-listener', function() {