fix bug
Merge pull request #7 from Nazgard/tests

fix bug