Fix tests
Fix tests build
0.2.1