Fix #12 - Make the entity reference field return entities instead of custom logic.