2 files changed, 14 insertions, 10 deletions

diff --git a/spec/outputs/5.1/attrib.lua b/spec/outputs/5.1/attrib.lua
index a156e84..bda24bc 100644
--- a/spec/outputs/5.1/attrib.lua
+++ b/spec/outputs/5.1/attrib.lua
@@ -136,9 +136,11 @@ do
         end
         local b
         if not false then
-                if x then
-                        b = 1
-                end
+                b = ((function()
+                        if x then
+                                return 1
+                        end
+                end)())
         end
         local _close_0
         if (function()
@@ -164,10 +166,12 @@ do
         end)(pcall(function()
                 local c
                 if true then
-                        local _exp_0 = x
-                        if "abc" == _exp_0 then
-                                c = 998
-                        end
+                        c = ((function()
+                                local _exp_0 = x
+                                if "abc" == _exp_0 then
+                                        return 998
+                                end
+                        end)())
                 end
                 local d
                 if (function()
diff --git a/spec/outputs/5.1/try_catch.lua b/spec/outputs/5.1/try_catch.lua
index d4c80c1..efd92c6 100644
--- a/spec/outputs/5.1/try_catch.lua
+++ b/spec/outputs/5.1/try_catch.lua
@@ -8,10 +8,10 @@ local _anon_func_2 = function(tb)
         return tb.func()
 end
 local _anon_func_3 = function(tb)
-        return tb.func()
+        return (tb.func())
 end
 local _anon_func_4 = function(tb)
-        return tb:func(1, 2, 3)
+        return (tb:func(1, 2, 3))
 end
 local _anon_func_5 = function(tb)
         return tb.func(1)
@@ -64,7 +64,7 @@ f = function()
                 print("OK")
         end
         if xpcall(function()
-                return func(1)
+                return (func(1))
         end, function(err)
                 return print(err)
         end) then