fixed tests to work with new api