update API in tests