Fixed tests.
Merge pull request #18 from matiasb/fix-tests

Fixed tests.