Small fix
Merge pull request #2 from piton4eg/patch-1

Small fix