fix tests
Merge pull request #219 from dimagi/tests

fix tests