Make unhighlight box actually work