Fixed error in tests
Merge pull request #3 from anowak/tests-fix

Fixed error in tests