Fix tests
0.5.1