Fix tests.
Merge pull request #50 from b17/testing

Fix tests.