Quick fix to failing tests
Merge pull request #45 from aviks/master

Quick fix to failing tests