Flatten correctly

danschultzer · Apr 25, 2018 · fb1bbbc · fb1bbbc
1 parent b50bbd8
commit fb1bbbc
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/premailex/html_to_plain_text.ex b/lib/premailex/html_to_plain_text.ex
@@ -141,9 +141,10 @@ defmodule Premailex.HTMLToPlainText do
     |> Enum.join(" ")
   end
 
-  defp flatten_table_body(list) when is_list(list), do: Enum.map(list, &flatten_table_body/1)
-  defp flatten_table_body({"tbody", [], table_cells}), do: table_cells
-  defp flatten_table_body(elem), do: elem
+  defp flatten_table_body(elements), do: Enum.flat_map(elements, &remove_tbody/1)
+
+  defp remove_tbody({"tbody", [], table_cells}), do: table_cells
+  defp remove_tbody(elem), do: [elem]
 
   defp wordwrap(text) do
     text