[ref #4] Fix unittest, wasn't updated with last interface change