fix: fix unit test

Fixes #17
Merge pull request #19 from thaiat/fix-unit-test

fix: fix unit test