Update test to work with v0.2.x