fix issue #5
Merge pull request #6 from Dav1dde/issue5

fix issue #5