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);
   }