Fix unit tests for 0.3.0 release.