Hotfix: fix unit tests
Merge pull request #28 from openplans/staging

Hotfix: fix unit tests