Source highlighting fix