Fix the selected row class reset after toggle column bug.