Return list when mapping over ReadResult

1 files changed, 2 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java b/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java
index beff7f7..9efd747 100644
--- a/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java
+++ b/src/main/java/org/zwobble/mammoth/internal/docx/ReadResult.java
@@ -63,9 +63,9 @@ public class ReadResult {
         this.warnings = warnings;
     }
 
-    public ReadResult map(Function<List<DocumentElement>, DocumentElement> function) {
+    public ReadResult map(Function<List<DocumentElement>, List<DocumentElement>> function) {
         return new ReadResult(
-            list(function.apply(elements)),
+            function.apply(elements),
             extra,
             warnings);
     }