Adjust test to new api