update test to follow recent changes.