updated test for the new api