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

Fixed error in tests