updating tests to the new API