Have UnmergedTableCell implement HasChildren

1 files changed, 11 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/StatefulBodyXmlReader.java b/src/main/java/org/zwobble/mammoth/internal/docx/StatefulBodyXmlReader.java
index 680bc92..161ef91 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/StatefulBodyXmlReader.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/StatefulBodyXmlReader.java
@@ -583,7 +583,7 @@ class StatefulBodyXmlReader {
             .orElse(false);
     }
 
-    private static class UnmergedTableCell implements DocumentElement {
+    private static class UnmergedTableCell implements DocumentElement, HasChildren<UnmergedTableCell> {
         private final boolean vmerge;
         private final int colspan;
         private final List<DocumentElement> children;
@@ -595,6 +595,16 @@ class StatefulBodyXmlReader {
         }
 
         @Override
+        public List<DocumentElement> getChildren() {
+            return children;
+        }
+
+        @Override
+        public UnmergedTableCell replaceChildren(List<DocumentElement> newChildren) {
+            return new UnmergedTableCell(this.vmerge, this.colspan, newChildren);
+        }
+
+        @Override
         public <T, U> T accept(DocumentElementVisitor<T, U> visitor, U context) {
             return visitor.visit(new TableCell(1, colspan, children), context);
         }