update test to follow api chages