Fix the bug in demo
Merge pull request #1 from zedzhao/dev

Fix the bug in demo