diff --git a/tests/test_recursive_parents.py b/tests/test_recursive_parents.py index 76178cee37db9ebeb2d249837bdfde80baa58141..460be3c4999fafc391120587fc7d8ece458ba9b5 100644 --- a/tests/test_recursive_parents.py +++ b/tests/test_recursive_parents.py @@ -137,8 +137,8 @@ def test_entity_has_parent(): assert not c.has_parent(fake_C_id, check_name=False, check_id=True) - fake_B_name_id = RecordType(name="TestTypeB", id=B.id) - fake_C_name_id = RecordType(name="not C", id=C.id * 5) + fake_B_name_id = db.RecordType(name="TestTypeB", id=B.id) + fake_C_name_id = db.RecordType(name="not C", id=C.id * 5) assert c.has_parent(fake_B_name_id, check_name=True, check_id=True)