Fixed small bug in search