small fix
Merge pull request #3 from rajatkapoor/fix

small fix