- updating tests for new api