fix demo bug

fix demo bug