Read endnote references

1 files changed, 5 insertions, 3 deletions

diff --git a/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java b/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java
index 5fe6f80..c8e1785 100644
--- a/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java
+++ b/src/main/java/org/zwobble/mammoth/docx/BodyXmlReader.java
@@ -55,7 +55,9 @@ public class BodyXmlReader {
             case "w:bookmarkStart":
                 return readBookmark(element);
             case "w:footnoteReference":
-                return readNoteReference(element);
+                return readNoteReference(NoteType.FOOTNOTE, element);
+            case "w:endnoteReference":
+                return readNoteReference(NoteType.ENDNOTE, element);
 
             case "w:ins":
             case "w:smartTag":
@@ -214,9 +216,9 @@ public class BodyXmlReader {
         }
     }
 
-    private ReadResult readNoteReference(XmlElement element) {
+    private ReadResult readNoteReference(NoteType noteType, XmlElement element) {
         String noteId = element.getAttribute("w:id");
-        return success(new NoteReference(noteId));
+        return success(new NoteReference(noteType, noteId));
     }
 
     private Optional<String> readVal(XmlElementLike element, String name) {