Little bug fix
Merge pull request #67 from Kenny9999/master

Little bug fix