I have the same problem, tried erverything I can but still didn't find a good solution until now.
Before this, if I want to do the alt-click/shift-click stuff, I have to replace my buttons with images. But doing so, I have to do the highlighting manually while my mouse is on the image. This is very fraustrating to work with. Your solution is great!
And can @Adobe fix this nasty bug so everybody can develope scripts better?