From d2d48e9f7f3dc5835c50f7999ff4c667d56725ab Mon Sep 17 00:00:00 2001
From: Sebastian McKenzie <sebmck@gmail.com>
Date: Mon, 2 Nov 2015 07:05:00 +0000
Subject: [PATCH] fix bug where async class methods weren't having their
 `await`s converted to `yield`s in the async to generator helper - fixes #2661

---
 packages/babel-helper-remap-async-to-generator/src/index.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/babel-helper-remap-async-to-generator/src/index.js b/packages/babel-helper-remap-async-to-generator/src/index.js
index ff19aa9133..1e84a4fec7 100644
--- a/packages/babel-helper-remap-async-to-generator/src/index.js
+++ b/packages/babel-helper-remap-async-to-generator/src/index.js
@@ -46,8 +46,6 @@ function plainFunction(path: NodePath, callId: Object) {
   node.async = false;
   node.generator = true;
 
-  path.traverse(awaitVisitor);
-
   let built = t.callExpression(callId, [node]);
   let container = buildWrapper({
     FUNCTION: built,
@@ -93,6 +91,8 @@ export default function (path: NodePath, callId: Object) {
   let node = path.node;
   if (node.generator) return;
 
+  path.traverse(awaitVisitor);
+
   if (path.isClassMethod()) {
     return classMethod(path, callId);
   } else {