update tests based on api changes