Fix test with last api changes