Update to work with the shadow dom