Fixing search

Search is not working, just fix it.