fix unit tests
Merge pull request #65 from markcowl/fixunits

fix unit tests