fix: use with babel-7 (#21)

jsbin · Dec 1, 2018 · 893ff8f · 893ff8f
1 parent 4cd640f
commit 893ff8f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/index.js b/lib/index.js
@@ -79,7 +79,8 @@ module.exports = (timeout = 100, extra = null) => {
   }
 
   return ({ types: t, transform }) => {
-    const node = extra ? transform(extra).ast.program.body[0] : null;
+    const node = extra ?
+      transform(extra, { ast: true }).ast.program.body[0] : null;
 
     let callback = null;
     if (t.isExpressionStatement(node)) {