fix unit tests to match new behavior
Merge pull request #35 from MichaelBlume/fix-tests

fix unit tests to match new behavior