set generator on correct node #252

2015-01-02 02:55:55 +11:00
parent 706626f79a
commit af21c52cc6
1 changed files with 1 additions and 1 deletions
--- 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);
  }