1 files changed, 2 insertions, 1 deletions
diff --git a/test/test-document-util.js b/test/test-document-util.js
index 99eea3ab..63a13581 100644
--- a/test/test-document-util.js
+++ b/test/test-document-util.js
@@ -19,6 +19,7 @@ const fs = require('fs');
 const path = require('path');
 const assert = require('assert');
 const {JSDOM} = require('jsdom');
+const {testMain} = require('../dev/util');
 const {VM} = require('../dev/vm');
 
 
@@ -238,4 +239,4 @@ async function main() {
 }
 
 
-if (require.main === module) { main(); }
+if (require.main === module) { testMain(main); }