updated test with updated API