Fix tests
0.2.2