diff --git a/tests/test_query.py b/tests/test_query.py
index 604871ded2d17d5fbe6cecbc27761b5be88ab524..269e58de006fcc5411976e9dbbd023074db7b1fb 100644
--- a/tests/test_query.py
+++ b/tests/test_query.py
@@ -32,7 +32,7 @@ import random
 
 import caosdb as db
 from caosdb.connection.connection import get_connection
-from linkahead.utils import get_entity_by_name, get_entity_by_id, get_entity_by_path
+from linkahead.utils.get_entity import get_entity_by_name, get_entity_by_id, get_entity_by_path
 from caosdb.exceptions import EmptyUniqueQueryError, TransactionError
 from lxml import etree
 from pytest import mark, raises