tests: fix broken tests

It is broken because the output is changed.
Merge pull request #28 from unihorn/116

tests: fix broken tests