fix demo

fix the demo