fix: fix unit test

Fixes #17