fix tests
0.6.1