fix bug of demo