diff --git a/luamml-amsmath.lua b/luamml-amsmath.lua
index 453b543..eacb91f 100644
--- a/luamml-amsmath.lua
+++ b/luamml-amsmath.lua
@@ -33,7 +33,7 @@ lua.get_functions_table()[funcid] = function()
     spacing[#spacing+1] = n.width == 0 and '0' or '.8em'
   end
   mml_table.columnspacing = table.concat(spacing, ' ', 2, #spacing-2)
-  save_result(mml_table, 0)
+  save_result(mml_table, true)
 end
 
 local last_tag
diff --git a/luamml-tex.lua b/luamml-tex.lua
index c76d1f7..b0ededa 100644
--- a/luamml-tex.lua
+++ b/luamml-tex.lua
@@ -38,8 +38,8 @@ end
 local mlist_buffer
 local mlist_result
 
-local function save_result(xml, style)
-  mlist_result = write_xml(make_root(xml, style))
+local function save_result(xml, display)
+  mlist_result = write_xml(make_root(xml, display and 0 or 2))
   if tex.count.tracingmathml > 1 then
     texio.write_nl(mlist_result .. '\n')
   end
@@ -64,7 +64,7 @@ luatexbase.add_to_callback('pre_mlist_to_hlist_filter', function(mlist, style)
   end
   local xml = process_mlist(new_mlist, style == 'display' and 0 or 2)
   if flag & 2 == 0 then
-    save_result(xml, (style == 'display' or flag & 1 == 1) and 0 or 2)
+    save_result(xml, style == 'display' or flag & 1 == 1)
   else
     assert(style == 'text')
     local startmath = tex.nest.top.tail