Updated the unit tests to v0.3.2