test unit: follow API change