update test to match API change