Update test to code changes.