From a0e9d4baf9c100bfb0034659095ebc6a0c07839e Mon Sep 17 00:00:00 2001
From: Valeriy S <ValeraS@users.noreply.github.com>
Date: Wed, 31 Oct 2018 15:22:30 +0300
Subject: [PATCH] fix: use with babel-7

---
 lib/index.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/index.js b/lib/index.js
index bf0ab1e0..bfeae59e 100644
--- 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)) {