update test to use new APIs