updated test for new api