add tests
bugfix
v0.1.0