Quick patching of a few things to allow UIElementButton to operate as
intended.