fix JuliaStaging/GeneralizedGenerated.jl#46

Author: thautwarm

Project.toml

@@ -1,16 +1,16 @@
 name = "JuliaVariables"
 uuid = "b14d175d-62b4-44ba-8fb7-3064adc8c3ec"
 authors = ["thautwarm"]
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 MLStyle = "d8e11817-5142-5d16-987a-aa16d5891078"
 NameResolution = "71a1bf82-56d0-4bbc-8a3c-48b961074391"
 
 [compat]
-julia = "1"
-MLStyle = "^0.3.1"
+MLStyle = "^0.3.1, ^0.4.1"
 NameResolution = "^0.1.3"
+julia = "1"
 
 [extras]
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

src/SimplifyEx.jl

@@ -1,13 +1,18 @@
 using MLStyle
 
+_wrap_block(ex::Expr) = ex
+_wrap_block(o) = Expr(:block, o)
+_simplify_block(ex) = _wrap_block(simplify_ex(ex))
+
+
 function simplify_ex(ex::Expr)
     @match ex begin
         Expr(:for, Expr(:block, assigns...), body) =>
-            foldr(assigns, init=simplify_ex(body)) do ass, last
+            foldr(assigns, init=_simplify_block(body)) do ass, last
                 Expr(:for, simplify_ex(ass), last)
             end
         Expr(:let, Expr(:block, assigns...), body) =>
-            foldr(assigns, init=simplify_ex(body)) do ass, last
+            foldr(assigns, init=_simplify_block(body)) do ass, last
                 Expr(:let, simplify_ex(ass), last)
             end
         Expr(:(=), lhs && Expr(:call, _...), rhs) =>