Fix core/test due to the api change.