Fixed the tests for v0.1.7