fix test to match current API