Fix search box bug