Properly grab the entity ID.