updated test to new API