lib/hamlit/compiler/script_compiler.rb in hamlit-2.2.0 vs lib/hamlit/compiler/script_compiler.rb in hamlit-2.2.1

@@ -71,13 +71,12 @@
            [:newline],
            [:code, ')'.freeze],
           ]
         else
           [:multi,
-           [:code, "#{var} = #{node.value[:text]}"],
-           [:newline],
-           yield(node),
-           [:code, 'end'.freeze],
+           [:block, "#{var} = #{node.value[:text]}",
+            [:multi, [:newline], yield(node)],
+           ],
           ]
         end
       end
 
       def compile_script_result(result, node)