Missed something in the highlight alogirth, this fixes it