diff --git a/packages/babel/src/api/node.js b/packages/babel/src/api/node.js
index 1b7dd84f73..04d75d825b 100644
--- a/packages/babel/src/api/node.js
+++ b/packages/babel/src/api/node.js
@@ -91,5 +91,11 @@ export function parse(code, opts = {}) {
     opts.features[key] = true;
   }
 
-  return babylon.parse(code, opts).program;
+  var ast = babylon.parse(code, opts)
+
+  if (opts.onToken) {
+    opts.onToken.push(...ast.tokens);
+  }
+
+  return ast.program;
 }