Fixing the condition
Merge pull request #160 from payshah/master

Fixing the condition