fix tests
2.0.2