Fix incorrect import in tests

diff --git a/tests/test_roundtrip.py b/tests/test_roundtrip.py
index ca50105ae013d1ccad44cdfba7818bd4552a8b02..fd53e501589d9eeb48e9139b87e38c935b12c947 100644 (file)
--- a/tests/test_roundtrip.py
+++ b/tests/test_roundtrip.py
@@ -5,7 +5,7 @@ from difflib import unified_diff
 import pytest
 import lxml.etree
 
-from drafthorse.models.document import Invoice
+from drafthorse.models.document import Document
 from drafthorse.utils import prettify, validate_xml
 
 samples = os.listdir(os.path.join(os.path.dirname(__file__), "samples"))
@@ -31,7 +31,7 @@ def test_sample_roundtrip(filename):
     validate_xml(xmlout=origxml, schema=schema)
 
     # Parse the sample file into our internal python structure
-    doc = Invoice.parse(origxml)
+    doc = Document.parse(origxml)
 
     # Validate output XML and render a diff for debugging
     # skip first line (namespace order…)