[TEST] update test for new API