fix unit tests
0.3.0