fix demo
fix bug.