Fixing tests that use the search box