Fix broken tests.
Merge pull request #8 from tylerdooling/master

Fix broken tests.