A small fix
Merge pull request #19 from Awesoham/patch-1

A small fix