Read style name for paragraphs out of source document

1 files changed, 6 insertions, 4 deletions

diff --git a/mammoth/docx/styles_xml.py b/mammoth/docx/styles_xml.py
index c220740..ab3cc0f 100644
--- a/mammoth/docx/styles_xml.py
+++ b/mammoth/docx/styles_xml.py
@@ -23,16 +23,18 @@ def read_styles_xml_element(element):
     
     for style_element in element.find_children("w:style"):
         style = _read_style_element(style_element)
-        styles[style.element_type][style.style_id] = style
+        element_type = style_element.attributes["w:type"]
+        style_set = styles.get(element_type)
+        if style_set is not None:
+            style_set[style.style_id] = style
     
     return Styles(paragraph_styles, character_styles)
 
 
 def _read_style_element(element):
-    element_type = element.attributes["w:type"]
     style_id = element.attributes["w:styleId"]
     name = element.find_child("w:name").attributes["w:val"]
-    return Style(element_type=element_type, style_id=style_id, name=name)
+    return Style(style_id=style_id, name=name)
 
 
-Style = collections.namedtuple("Style", ["element_type", "style_id", "name"])
+Style = collections.namedtuple("Style", ["style_id", "name"])