Small fix
Merge pull request #24 from ale8530/master

Small fix