Small bugfix

Small bugfix
Merge pull request #3 from kapoor/patch-1

Small bugfix