workaround bug in ECMAScript specification (#4656)

closes #4655

1 files changed, 7 insertions, 5 deletions

diff --git a/lib/compress.js b/lib/compress.js
index b4f9c3a8..a2a28ab9 100644
--- a/lib/compress.js
+++ b/lib/compress.js
@@ -5017,20 +5017,22 @@ merge(Compressor.prototype, {
         return self;
     });
 
-    function trim_block(node, in_list) {
+    function trim_block(node, parent, in_list) {
         switch (node.body.length) {
           case 0:
             return in_list ? List.skip : make_node(AST_EmptyStatement, node);
           case 1:
             var stat = node.body[0];
-            if (!(stat instanceof AST_Const || stat instanceof AST_Let)) return stat;
+            if (stat instanceof AST_Const || stat instanceof AST_Let) return node;
+            if (parent instanceof AST_IterationStatement && stat instanceof AST_LambdaDefinition) return node;
+            return stat;
         }
         return node;
     }
 
     OPT(AST_BlockStatement, function(self, compressor) {
         self.body = tighten_body(self.body, compressor);
-        return trim_block(self);
+        return trim_block(self, compressor.parent());
     });
 
     function drop_rest_farg(fn, compressor) {
@@ -6123,7 +6125,7 @@ merge(Compressor.prototype, {
             }
         }, function(node, in_list) {
             if (node instanceof AST_BlockStatement) {
-                return trim_block(node, in_list);
+                return trim_block(node, tt.parent(), in_list);
             } else if (node instanceof AST_For) {
                 // Certain combination of unused name + side effect leads to invalid AST:
                 //    https://github.com/mishoo/UglifyJS/issues/44
@@ -7282,7 +7284,7 @@ merge(Compressor.prototype, {
                     break;
                 }
             }
-            self.body = trim_block(self.body);
+            self.body = trim_block(self.body, compressor.parent());
         }
         if (self.body instanceof AST_EmptyStatement) return make_node(AST_For, self, self).optimize(compressor);
         if (self.body instanceof AST_SimpleStatement) return make_node(AST_For, self, {