fix tests
0.1.1