From dac4263900fa7edf36c3b03528407df518455661 Mon Sep 17 00:00:00 2001
From: Raphael Michel <michel@rami.io>
Date: Wed, 3 Aug 2022 12:26:21 +0200
Subject: [PATCH] Fix incorrect import in tests

---
 tests/test_roundtrip.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_roundtrip.py b/tests/test_roundtrip.py
index ca50105..fd53e50 100644
--- 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…)
-- 
2.47.3