Set focus on undotree toggle