fix demo bug.