Small fix
Merge pull request #36 from ale8530/master

Small fix