Fixed a minor bug which caused the highlighting to malfunction