fix tests
Merge pull request #10 from abarabash89/fix_tests

fix tests