fix test failure
Merge pull request #24 from blturner/fix-tests

fix test failure