fixed demo bug