4 files changed, 7 insertions, 7 deletions
diff --git a/testsuite/pyunit/dom/Expressions.py b/testsuite/pyunit/dom/Expressions.py
index 3a4f658af..a7afb30ba 100644
--- a/testsuite/pyunit/dom/Expressions.py
+++ b/testsuite/pyunit/dom/Expressions.py
@@ -5,8 +5,8 @@ from unittest import TestCase
 from pyGHDL.dom.DesignUnit import Package
 
 from pyGHDL.dom import Expression
-from pyGHDL.dom.Misc       import Design, Document
-from pyGHDL.dom.Symbol import SimpleObjectSymbol
+from pyGHDL.dom.NonStandard       import Design, Document
+from pyGHDL.dom.Symbol import SimpleObjectOrFunctionCallSymbol
 from pyGHDL.dom.Object import Constant
 from pyGHDL.dom.Expression import InverseExpression
 
@@ -39,7 +39,7 @@ class Expressions(TestCase):
         item: Constant = package.DeclaredItems[0]
         default: Expression = item.DefaultExpression
         self.assertTrue(isinstance(default, InverseExpression))
-        self.assertTrue(isinstance(default.Operand, SimpleObjectSymbol))
+        self.assertTrue(isinstance(default.Operand, SimpleObjectOrFunctionCallSymbol))
         self.assertTrue(default.Operand.SymbolName == "true")
 
     # def test_Aggregare(self):
@@ -66,5 +66,5 @@ class Expressions(TestCase):
     #     item: Constant = package.DeclaredItems[0]
     #     default: Expression = item.DefaultExpression
     #     self.assertTrue(isinstance(default, InverseExpression))
-    #     self.assertTrue(isinstance(default.Operand, SimpleObjectSymbol))
+    #     self.assertTrue(isinstance(default.Operand, SimpleObjectOrFunctionCallSymbol))
     #     self.assertTrue(default.Operand.SymbolName == "true")
diff --git a/testsuite/pyunit/dom/Literals.py b/testsuite/pyunit/dom/Literals.py
index 7eb80abaa..c542ebfe5 100644
--- a/testsuite/pyunit/dom/Literals.py
+++ b/testsuite/pyunit/dom/Literals.py
@@ -2,7 +2,7 @@ from pathlib  import Path
 from textwrap import dedent
 from unittest import TestCase
 
-from pyGHDL.dom.Misc       import Design, Document
+from pyGHDL.dom.NonStandard       import Design, Document
 from pyGHDL.dom.Object import Constant
 from pyGHDL.dom.Literal import IntegerLiteral
 
diff --git a/testsuite/pyunit/dom/SimpleEntity.py b/testsuite/pyunit/dom/SimpleEntity.py
index 252032f9e..8199fe7cc 100644
--- a/testsuite/pyunit/dom/SimpleEntity.py
+++ b/testsuite/pyunit/dom/SimpleEntity.py
@@ -1,7 +1,7 @@
 from pathlib  import Path
 from unittest import TestCase
 
-from pyGHDL.dom.Misc       import Design, Library, Document
+from pyGHDL.dom.NonStandard       import Design, Document
 
 
 if __name__ == "__main__":
diff --git a/testsuite/pyunit/dom/SimplePackage.py b/testsuite/pyunit/dom/SimplePackage.py
index 212e0402d..5b16e74b8 100644
--- a/testsuite/pyunit/dom/SimplePackage.py
+++ b/testsuite/pyunit/dom/SimplePackage.py
@@ -1,7 +1,7 @@
 from pathlib  import Path
 from unittest import TestCase
 
-from pyGHDL.dom.Misc       import Design, Document
+from pyGHDL.dom.NonStandard       import Design, Document
 
 
 if __name__ == "__main__":