fix tests
0.2.1