Fixes the tests
Bumps to v0.1.6