Fixing broken tests
Release v0.2.1