fix test case compile error to adapt new API change.