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

augmented-go/augmented-go,augmented-go/augmented-go,augmented-go/augmented-go