Fixed tests.
Merge pull request #25 from tuomasvapaavuori/test_fix

Fixed tests.