Update to work properly with the Shadow DOM

Fixes #8