fix bug with demo