You are right I should have put this on the button. 
I missed that this was previously assigned to `getShadowDomElement('button')`