fix demo bug