Update tests to account for wordsearch changes