fix unit test after interface change