Fix mapping for search highlight