fix tests.
Merged pull request #110