update tests to new api