FIx test failures in 0.2.7

This PR attempts to fix the failing tests mentioned in #34.