Updating tests for my pull request.