fix demo
Merge pull request #9 from willpragnell/demo

Fix demo