fix tests
Merge pull request #50 from dimagi/fix-tests

fix tests