fix broken unit tests
Merge pull request #85 from dcrissman/fix-failing-unit-tests

fix broken unit tests