Fixed the non highlighting after the first click

Title, commented a line