in my tests the last version highlights the selected button properly, doesn't lose focus when you change states and doesn't require mouse click to activate key navigation
other than the losing focus on state change, everything else should be handled by the eventsystem, I don't think it has to do with the action but with how it is set up