1 files changed, 5 insertions, 7 deletions

diff --git a/spec/outputs/assign.lua b/spec/outputs/assign.lua
index f889865..89c5f8a 100644
--- a/spec/outputs/assign.lua
+++ b/spec/outputs/assign.lua
@@ -36,17 +36,15 @@ local x
 do
         local f = getHandler()
         if f then
-                do
+                x = ((function()
                         f()
-                        x = 123
-                end
+                        return 123
+                end)())
         end
 end
 local _anon_func_0 = function(print)
-        do
-                print(123)
-                return { }
-        end
+        print(123)
+        return { }
 end
 return _(function()
         setmetatable(a, _anon_func_0(print))