Fixed issues in test code (fixes #3)
Merge pull request #4 from vania-pooh/master

Fixed issues in test code (fixes #3)