Adjust test to match new code.