Use table style mappings during document conversion

1 files changed, 3 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java b/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
index 8b15106..3f3576f 100644
--- a/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
+++ b/src/main/java/org/zwobble/mammoth/internal/conversion/DocumentToHtml.java
@@ -226,8 +226,9 @@ public class DocumentToHtml {
 
             @Override
             public List<HtmlNode> visit(Table table, Context context) {
-                List<HtmlNode> children = generateTableChildren(table, context);
-                return list(Html.element("table", children));
+                HtmlPath mapping = styleMap.getTableHtmlPath(table)
+                    .orElse(HtmlPath.element("table"));
+                return mapping.wrap(() -> generateTableChildren(table, context)).get();
             }
 
             private List<HtmlNode> generateTableChildren(Table table, Context context) {