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 {