diff --git a/tests/test_query.py b/tests/test_query.py
index 81e1501bd5b3fabad4bed4f8643011d17924220c..1a06c080b16fc63a5b1323ca8a6f0e7834c2648c 100644
--- a/tests/test_query.py
+++ b/tests/test_query.py
@@ -29,12 +29,12 @@
 
 import os
 import random
-import caosdb as db
 
-from pytest import mark, raises
+import caosdb as db
 from caosdb.connection.connection import get_connection
 from caosdb.exceptions import EmptyUniqueQueryError, TransactionError
 from lxml import etree
+from pytest import mark, raises
 
 
 def setup_module():
@@ -93,6 +93,8 @@ def test_query_with_reference_by_parent():
         unique=True).id
     assert exp_rec.id == db.execute_query(
         "FIND TestExperiment .-> " + str(sp_rec.id), unique=True).id
+    assert exp_rec.id == db.execute_query("FIND ENTITY " + str(exp_rec.id), unique=True).id
+    assert exp_rec.id == db.execute_query("FIND RECORD " + str(exp_rec.id), unique=True).id
     assert exp_rec.id == db.execute_query(
         "FIND TestExperiment .-> TestSpecialProtocolLog1", unique=True).id
     assert exp_rec.id == db.execute_query(