diff --git a/unittests/test_apiutils.py b/unittests/test_apiutils.py
index b9256fcd476d149772c3bce2b238522c62b6a53a..43d4955e4ffb9c3646760475a4ba664fd8e657db 100644
--- a/unittests/test_apiutils.py
+++ b/unittests/test_apiutils.py
@@ -82,7 +82,8 @@ def test_resolve_reference():
     resolve_reference(prop)
     resolve_reference(prop_list)
     resolve_reference(prop_list2)
-    assert prop.value.id == db.Record(id=100).id
+    assert prop.value.id == 100
+    assert isinstance(prop.value, db.Entity) == True
 
     prop_list_ids = []
     for i in prop_list.value: