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() {