Update tests for v0.9.0