4 files changed, 24 insertions, 12 deletions

diff --git a/spec/outputs/syntax.lua b/spec/outputs/syntax.lua
index cfe63eb..5fd1821 100644
--- a/spec/outputs/syntax.lua
+++ b/spec/outputs/syntax.lua
@@ -40,12 +40,16 @@ end
 _ = function()
         return 1, 2, 34
 end
-return 5 + function()
-        return 4 + 2
+do
+        return 5 + function()
+                return 4 + 2
+        end
+end
+do
+        return 5 + (function()
+                return 4
+        end) + 2
 end
-return 5 + (function()
-        return 4
-end) + 2
 print(5 + function()
         _ = 34
         return good(nads)
diff --git a/spec/outputs/unicode/syntax.lua b/spec/outputs/unicode/syntax.lua
index 9ea8f68..ea97bb9 100644
--- a/spec/outputs/unicode/syntax.lua
+++ b/spec/outputs/unicode/syntax.lua
@@ -40,12 +40,16 @@ end
 _ = function()
         return 1, 2, 34
 end
-return 5 + function()
-        return 4 + 2
+do
+        return 5 + function()
+                return 4 + 2
+        end
+end
+do
+        return 5 + (function()
+                return 4
+        end) + 2
 end
-return 5 + (function()
-        return 4
-end) + 2
 _u6253_u5370(5 + function()
         _ = 34
         return _u597d(_u7403)
diff --git a/spec/outputs/unicode/with.lua b/spec/outputs/unicode/with.lua
index 7a5ba00..cfad264 100644
--- a/spec/outputs/unicode/with.lua
+++ b/spec/outputs/unicode/with.lua
@@ -123,7 +123,9 @@ do
         local _
         _ = function()
                 local _with_0 = _u55e8
-                return _with_0.a, _with_0.b
+                do
+                        return _with_0.a, _with_0.b
+                end
         end
 end
 do
diff --git a/spec/outputs/with.lua b/spec/outputs/with.lua
index b2a1c3b..d880d1e 100644
--- a/spec/outputs/with.lua
+++ b/spec/outputs/with.lua
@@ -121,7 +121,9 @@ do
         local _
         _ = function()
                 local _with_0 = hi
-                return _with_0.a, _with_0.b
+                do
+                        return _with_0.a, _with_0.b
+                end
         end
 end
 do