Not retaining the text highlight in jtextpane

Question

I have a jtextpane and a button in a frame. when the user selects some text in textpane and clicks on the button, a word gets appended to the selected text and also gets highlighted. It works fine when the button is clicked but the when the focus changes, the highlighted text doesnt retain the color, but retains the appended text. is there anything i am missing in the code to retain the highlight?
private Highlighter.HighlightPainter yellowPainter = new DefaultHighlighter.DefaultHighlightPainter( Color.YELLOW);
public void actionPerformed(ActionEvent ae){
        JEditorPane textForComments = getEditor(ae);
                if (textForComments != null) {
                int caretPos1 = textForComments.getSelectionEnd();
                int selStart = textForComments.getSelectionStart();
               textForComments.getDocument().insertString(caretPos1, "ID", null);
               Object highlightTag = textForComments.getHighlighter().addHighlight(selStart ,caretPos1 , yellowPainter);
}

Answer 1

Selected text is only highlighted when focus is on the text component. Since you clicked on the button the button now has focus. You need to invoke
textForComments.requestFocusInWindow();