fix test to match new API