fix unit tests
Merge pull request #17 from e-schultz/chore-fix-tests

fix unit tests