fix test to match actual api