Fix search highlight