diff --git a/tests/func/import/diamond/test_diamond.py b/tests/func/import/diamond/test_diamond.py
index 8913dd8..6f67415 100644
--- a/tests/func/import/diamond/test_diamond.py
+++ b/tests/func/import/diamond/test_diamond.py
@@ -55,7 +55,7 @@ def test_diamond_import_resolving_and_model_creation():
     assert len(module.components) == 1
 
     component = module.components[0]
-    assert type(component) == type(packageComponent)
+    assert type(component) is type(packageComponent)
     assert component.name == 'myComponent'
     assert len(component.slots) == 2
 
diff --git a/tests/func/persistence/compare_table/test_compare_table.py b/tests/func/persistence/compare_table/test_compare_table.py
index 876a7c6..596a65d 100644
--- a/tests/func/persistence/compare_table/test_compare_table.py
+++ b/tests/func/persistence/compare_table/test_compare_table.py
@@ -64,7 +64,7 @@ def test_diamond_import_resolving_and_model_creation():
     assert len(module.components) == 1
 
     component = module.components[0]
-    assert type(component) == type(packageComponent)
+    assert type(component) is type(packageComponent)
     assert component.name == 'myComponent'
     assert len(component.slots) == 2