From af21c52cc6fdcdca26211a200ca528e01cf54d83 Mon Sep 17 00:00:00 2001
From: Sebastian McKenzie <sebmck@gmail.com>
Date: Fri, 2 Jan 2015 02:55:55 +1100
Subject: [PATCH] set generator on correct node #252

---
 lib/6to5/transformation/transformers/es7-array-comprehension.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/6to5/transformation/transformers/es7-array-comprehension.js b/lib/6to5/transformation/transformers/es7-array-comprehension.js
index fd322a9d57..9e3d5bfff1 100644
--- a/lib/6to5/transformation/transformers/es7-array-comprehension.js
+++ b/lib/6to5/transformation/transformers/es7-array-comprehension.js
@@ -16,7 +16,7 @@ var build = function (node, file) {
   var body  = block.body;
 
   if (traverse.hasType(node, "YieldExpression", t.FUNCTION_TYPES)) {
-    container.generator = true;
+    container.callee.generator = true;
     container = t.yieldExpression(container, true);
   }