fix #8
Merge pull request #10 from b4rtek/bug

fix #8