Fix broken build
Merge pull request #24 from wshafer/test_fix

Fix broken build