Make load referenced entities more performant

Problem: Currently are all related entities loaded, which is no problem while having few entities, but gets more inperformant with a huge amount of entities.

Change to entity query.