fixing tests for new api