fix bug
Merge pull request #8 from Nazgard/tests

fix bug