fix tests
v0.6.0