Fix focus in button on click in buttons