[tests] updated to match new api