You could move this check into initHighlight and return early instead.