Updating test for refactor.