Fixing error in code
Merge pull request #912 from Tyriar/patch-2

Fixing error in code