test: follow API change