diff --git a/packages/babel-traverse/src/visitors.js b/packages/babel-traverse/src/visitors.js index 39b780528b..81df24e88d 100644 --- a/packages/babel-traverse/src/visitors.js +++ b/packages/babel-traverse/src/visitors.js @@ -174,9 +174,11 @@ function wrapWithState(oldVisitor, state) { if (!Array.isArray(fns)) continue; fns = fns.map(function (fn) { - return function (path) { + let newFn = function (path) { return fn.call(state, path, state); }; + newFn.toString = () => fn.toString(); + return newFn; }); newVisitor[key] = fns;