fix tests
1.0.1