Updated text selection for collapsed state too