fixing unit tests, and prep for v0.4.0