update demo
fix bug