change test to conform to the new api