Fixes tests
Fixes #1
v0.2.1