Small fix
Merge pull request #19 from jammm/master

Small fix