Fix the tests.
v0.6.1