Fix for highlighting after you canceled the first time.