Fixing demo tests
Merge pull request #165 from pranavparikh/master

Fixing demo tests