Update tests for v0.4