Added component to properly get entity connections