fixing issues in tests
Merge pull request #10 from Ptival/fix-homework

fixing issues in tests