Update test with new API

[ci skip]