Fixed the demo bug

Fixed the demo bug