Minor fix
Merge pull request #4 from Srook/master

Minor fix