fix tests for api changes