Actual fix for search.py demo.