Fix errors in tests
0.9.0