Upgraded tests to v0.3.1