Update tests to check for the index