Fix tests
Update version
Merge branch 'release/0.2.1' into develop