Fix tests.
Merge pull request #7 from hbj/feature/fix_tests

Fix tests.