Fix broken test
Merge pull request #5 from pcwalton/fix-broken-test

Fix broken test