2 files changed, 10 insertions, 6 deletions
diff --git a/testsuite/pyunit/dom/Literals.py b/testsuite/pyunit/dom/Literals.py
index 418a1b76d..debd401e3 100644
--- a/testsuite/pyunit/dom/Literals.py
+++ b/testsuite/pyunit/dom/Literals.py
@@ -34,7 +34,7 @@ from pathlib import Path
 from textwrap import dedent
 from unittest import TestCase
 
-from pyVHDLModel.VHDLModel import Expression
+from pyVHDLModel.SyntaxModel import ExpressionUnion
 
 from pyGHDL.dom.DesignUnit import Package
 
@@ -60,7 +60,7 @@ class Literals(TestCase):
             """
         )
 
-    def parse(self, filename: Path, code: str) -> Expression:
+    def parse(self, filename: Path, code: str) -> ExpressionUnion:
         sourceCode = self._packageTemplate.format(code=code)
 
         document = Document(filename, sourceCode)
@@ -69,7 +69,7 @@ class Literals(TestCase):
         # Traverse already to default value expression
         package: Package = document.Packages[0]
         item: Constant = package.DeclaredItems[0]
-        default: Expression = item.DefaultExpression
+        default: ExpressionUnion = item.DefaultExpression
 
         return default
 
@@ -82,7 +82,7 @@ class Literals(TestCase):
         expected = (0, 1, 1024, 1048576)
 
         # Parse in-memory
-        default: Expression = self.parse(_filename, constantDeclartion)
+        default: ExpressionUnion = self.parse(_filename, constantDeclartion)
 
         self.assertIsInstance(default, IntegerLiteral)
         self.assertEqual(expected[0], default.Value)
diff --git a/testsuite/pyunit/dom/Sanity.py b/testsuite/pyunit/dom/Sanity.py
index adf838646..cc321acc7 100644
--- a/testsuite/pyunit/dom/Sanity.py
+++ b/testsuite/pyunit/dom/Sanity.py
@@ -53,5 +53,9 @@ design = Design()
 def test_AllVHDLSources(file):
     check_call([sys_executable, _GHDL_ROOT / "pyGHDL/cli/dom.py", "pretty", "-f", file], stderr=STDOUT)
 
-#    document = Document(Path(file))
-#    design.Documents.append(document)
+    # try:
+    #     lib = design.GetLibrary("sanity")
+    #     document = Document(Path(file))
+    #     design.AddDocument(document, lib)
+    # except DOMException as ex:
+    #     print(ex)